Home
last modified time | relevance | path

Searched refs:test_unshard_async (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/fsdp/
H A Dtest_fsdp_comm.py299 def test_unshard_async(self, use_orig_params: bool): member in TestExplicitUnshard
/aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/
H A Dtest_fully_shard_comm.py1024 def test_unshard_async(self): member in TestFullyShardUnshardMultiProcess