Home
last modified time | relevance | path

Searched refs:CompileCounterWithBackend (Results 1 – 23 of 23) sorted by relevance

/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_aot_autograd.py302 cc = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
324 cc = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
403 cc = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
426 cc = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
459 cc = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
472 cc = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
513 cc = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
549 cc = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
562 cc = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
599 cc = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
[all …]
H A Dtest_higher_order_ops.py20 CompileCounterWithBackend,
138 cnt = CompileCounterWithBackend(backend)
176 cnt = CompileCounterWithBackend(backend)
461 cnt = CompileCounterWithBackend(backend)
609 cnt = CompileCounterWithBackend(backend)
1169 cnt = CompileCounterWithBackend(backend)
1290 cnt = CompileCounterWithBackend(backend)
1324 cnt = CompileCounterWithBackend(backend)
1411 cnt = CompileCounterWithBackend(backend)
1457 cnt = CompileCounterWithBackend(backend)
[all …]
H A Dtest_input_attr_tracking.py9 CompileCounterWithBackend,
301 counter = CompileCounterWithBackend(eager_and_record)
H A Dtest_activation_checkpointing.py17 from torch._dynamo.testing import CompileCounterWithBackend
461 cnt = CompileCounterWithBackend(backend)
490 cnt = CompileCounterWithBackend(backend)
516 cnt = CompileCounterWithBackend(backend)
1275 cnt = CompileCounterWithBackend("eager")
H A Dtest_decorators.py117 cnts = torch._dynamo.testing.CompileCounterWithBackend("eager")
139 cnts = torch._dynamo.testing.CompileCounterWithBackend("eager")
H A Dtest_modules.py1939 cnts = torch._dynamo.testing.CompileCounterWithBackend("eager")
1991 cnts = torch._dynamo.testing.CompileCounterWithBackend("eager")
2005 cnts = torch._dynamo.testing.CompileCounterWithBackend("eager")
2043 cnts = torch._dynamo.testing.CompileCounterWithBackend("eager")
2242 cc = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
2311 cc = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
H A Dtest_hooks.py663 cnts = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
738 cnts = torch._dynamo.testing.CompileCounterWithBackend(backend)
H A Dtest_autograd_function.py493 cnt = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
1062 cnt = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
H A Dtest_functions.py23 CompileCounterWithBackend,
2393 cnts = CompileCounterWithBackend(backend)
2440 cnts = CompileCounterWithBackend(backend)
2491 cnts = CompileCounterWithBackend(backend)
2539 cnts = CompileCounterWithBackend(backend)
H A Dtest_subclasses.py953 cnt = torch._dynamo.testing.CompileCounterWithBackend(backend)
1059 cnt = torch._dynamo.testing.CompileCounterWithBackend(backend)
1608 cnt = torch._dynamo.testing.CompileCounterWithBackend(backend)
H A Dtest_misc.py45 CompileCounterWithBackend,
2502 cnt = CompileCounterWithBackend("inductor")
5079 cnt = torch._dynamo.testing.CompileCounterWithBackend(backend)
5120 cnt = torch._dynamo.testing.CompileCounterWithBackend(backend)
10201 cnt = torch._dynamo.testing.CompileCounterWithBackend("eager")
H A Dtest_repros.py4304 cnt = torch._dynamo.testing.CompileCounterWithBackend(backend)
/aosp_15_r20/external/pytorch/test/distributed/_tensor/
H A Dtest_dtensor_compile.py280 cnt = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
319 cnt = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
336 cnt = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
359 cnt = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
535 cnt = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
556 cnt = torch._dynamo.testing.CompileCounterWithBackend("inductor")
708 cnt = torch._dynamo.testing.CompileCounterWithBackend("inductor")
784 cnt = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
835 cnt = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
/aosp_15_r20/external/pytorch/torch/utils/benchmark/utils/
H A Dcompile.py6 from torch._dynamo.testing import CompileCounterWithBackend
89 compile_counter_with_backend = CompileCounterWithBackend(backend)
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_dynamo_distributed.py667 cnt = torch._dynamo.testing.CompileCounterWithBackend("inductor")
1451 cnt = torch._dynamo.testing.CompileCounterWithBackend(backend)
1659 cnt = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
1702 cnt = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
1714 cnt = torch._dynamo.testing.CompileCounterWithBackend("eager")
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_control_flow.py183 cnt = torch._dynamo.testing.CompileCounterWithBackend("inductor")
593 cnt = torch._dynamo.testing.CompileCounterWithBackend("inductor")
H A Dtest_flex_attention.py13 from torch._dynamo.testing import CompileCounterWithBackend, normalize_gm
990 cnt = CompileCounterWithBackend("aot_eager")
H A Dtest_cuda_repro.py259 cnts = torch._dynamo.testing.CompileCounterWithBackend("inductor")
1063 cnts = torch._dynamo.testing.CompileCounterWithBackend("inductor")
H A Dtest_torchinductor.py35 CompileCounterWithBackend,
8875 cnts = CompileCounterWithBackend("inductor")
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dtesting.py188 class CompileCounterWithBackend: class
/aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/
H A Dtest_fully_shard_compile.py125 cnt = torch._dynamo.testing.CompileCounterWithBackend("aot_eager")
/aosp_15_r20/external/pytorch/test/export/
H A Dtest_torchbind.py1147 cnt = torch._dynamo.testing.CompileCounterWithBackend(backend)
/aosp_15_r20/external/pytorch/test/
H A Dtest_transformers.py38 from torch._dynamo.testing import CompileCounterWithBackend
3506 cnts = CompileCounterWithBackend("aot_eager")