Searched refs:test_unshard_async (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/fsdp/ | ||
| H A D | test_fsdp_comm.py | 299 def test_unshard_async(self, use_orig_params: bool): member in TestExplicitUnshard |
| /aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/ | ||
| H A D | test_fully_shard_comm.py | 1024 def test_unshard_async(self): member in TestFullyShardUnshardMultiProcess |