Home
last modified time | relevance | path

Searched refs:is_torchdynamo_compiling (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/tensor/parallel/
H A D_utils.py11 from torch._dynamo.external_utils import is_compiling as is_torchdynamo_compiling unknown
14 def is_torchdynamo_compiling(): # type: ignore[misc] function
29 if not is_torchdynamo_compiling():
/aosp_15_r20/external/pytorch/torch/distributed/fsdp/
H A D_runtime_utils.py317 if not torch.distributed._functional_collectives.is_torchdynamo_compiling():
420 if not torch.distributed._functional_collectives.is_torchdynamo_compiling():
673 if not torch.distributed._functional_collectives.is_torchdynamo_compiling():
739 if not torch.distributed._functional_collectives.is_torchdynamo_compiling():
854 if not torch.distributed._functional_collectives.is_torchdynamo_compiling():
1165 if torch.distributed._functional_collectives.is_torchdynamo_compiling():
1435 if torch.distributed._functional_collectives.is_torchdynamo_compiling():
1484 if torch.distributed._functional_collectives.is_torchdynamo_compiling():
1500 if torch.distributed._functional_collectives.is_torchdynamo_compiling():
1523 if not torch.distributed._functional_collectives.is_torchdynamo_compiling():
[all …]
H A D_exec_order_utils.py218 if not torch.distributed._functional_collectives.is_torchdynamo_compiling():
244 if not torch.distributed._functional_collectives.is_torchdynamo_compiling():
H A D_common_utils.py546 if torch.distributed._functional_collectives.is_torchdynamo_compiling():
H A D_flat_param.py916 if not torch.distributed._functional_collectives.is_torchdynamo_compiling():
2551 if not torch.distributed._functional_collectives.is_torchdynamo_compiling():
/aosp_15_r20/external/pytorch/torch/distributed/
H A D_functional_collectives.py23 def is_torchdynamo_compiling(): function
29 from torch.compiler import is_dynamo_compiling as is_torchdynamo_compiling unknown
35 def is_torchdynamo_compiling(): function
767 if not is_torchdynamo_compiling():
799 if is_torchdynamo_compiling():
H A Dutils.py180 if not torch.distributed._functional_collectives.is_torchdynamo_compiling():
200 if not torch.distributed._functional_collectives.is_torchdynamo_compiling():