Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/distributed/_composable/fully_shard/
H A Dtest_fully_shard_model_checkpoint.py70 _zero_model(load_composable, summon_full=False)
102 _zero_model(load_composable, summon_full=False)
160 _zero_model(load_model, zero_buffers=True, summon_full=False)
199 _zero_model(load_composable, summon_full=False)
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_fsdp.py145 summon_full=True, argument
148 ctx = FSDP.summon_full_params(model) if summon_full else nullcontext()