Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_torchinductor_dynamic_shapes.py243 @torch._dynamo.config.patch(capture_scalar_outputs=True)
278 @torch._dynamo.config.patch(capture_scalar_outputs=True)
287 @torch._dynamo.config.patch(capture_scalar_outputs=True)
296 capture_scalar_outputs=True, capture_dynamic_output_shape_ops=True
307 @torch._dynamo.config.patch(capture_scalar_outputs=True)
318 @torch._dynamo.config.patch(capture_scalar_outputs=True)
329 capture_scalar_outputs=True, capture_dynamic_output_shape_ops=True
339 capture_scalar_outputs=True, capture_dynamic_output_shape_ops=True
348 @torch._dynamo.config.patch(capture_scalar_outputs=True)
368 capture_scalar_outputs=True, capture_dynamic_output_shape_ops=True
[all …]
H A Dtest_unbacked_symints.py264 @torch._dynamo.config.patch(capture_scalar_outputs=True)
H A Dtest_auto_functionalize.py340 capture_scalar_outputs=True, capture_dynamic_output_shape_ops=True
H A Dtest_torchinductor.py4906 @torch._dynamo.config.patch(capture_scalar_outputs=True)
4929 @torch._dynamo.config.patch(capture_scalar_outputs=True)
4997 @torch._dynamo.config.patch(capture_scalar_outputs=True)
5011 @torch._dynamo.config.patch(capture_scalar_outputs=True)
5033 @torch._dynamo.config.patch(capture_scalar_outputs=True)
10425 @torch._dynamo.config.patch(capture_scalar_outputs=True)
10448 @torch._dynamo.config.patch(capture_scalar_outputs=True)
H A Dtest_triton_kernels.py1115 @torch._dynamo.config.patch(capture_scalar_outputs=True)
H A Dtest_aot_inductor.py2786 @torch._dynamo.config.patch(capture_scalar_outputs=True)
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_unspec.py456 @torch._dynamo.config.patch(capture_scalar_outputs=True)
530 @torch._dynamo.config.patch(capture_scalar_outputs=True)
639 @torch._dynamo.config.patch(capture_scalar_outputs=True)
686 @torch._dynamo.config.patch(capture_scalar_outputs=True)
H A Dtest_export.py468 @config.patch(capture_scalar_outputs=True)
514 @config.patch(capture_scalar_outputs=True)
538 @config.patch(capture_scalar_outputs=True)
562 @config.patch(capture_scalar_outputs=True)
975 @config.patch(capture_scalar_outputs=True)
1700 @config.patch(capture_scalar_outputs=True)
1738 @config.patch(capture_scalar_outputs=True)
1771 @config.patch(capture_scalar_outputs=True)
2059 @config.patch(capture_scalar_outputs=True)
2761 capture_scalar_outputs=True,
[all …]
H A Dtest_repros.py1347 fake_tensor_propagation=True, capture_scalar_outputs=True
1369 @torch._dynamo.config.patch(capture_scalar_outputs=False)
2239 @torch._dynamo.config.patch(capture_scalar_outputs=True)
3508 @torch._dynamo.config.patch(capture_scalar_outputs=True)
3954 @torch._dynamo.config.patch(capture_scalar_outputs=True)
4093 @torch._dynamo.config.patch(capture_scalar_outputs=True)
4405 @torch._dynamo.config.patch(capture_scalar_outputs=True)
H A Dtest_misc.py1377 @torch._dynamo.config.patch(capture_scalar_outputs=True)
1391 @torch._dynamo.config.patch(capture_scalar_outputs=True)
1410 @torch._dynamo.config.patch(capture_scalar_outputs=True)
8238 capture_scalar_outputs=True, capture_dynamic_output_shape_ops=True
8253 capture_scalar_outputs=True, capture_dynamic_output_shape_ops=True
8274 @torch._dynamo.config.patch(capture_scalar_outputs=True)
8285 @torch._dynamo.config.patch(capture_scalar_outputs=True)
8298 capture_scalar_outputs=True, capture_dynamic_output_shape_ops=True
8476 @torch._dynamo.config.patch(capture_scalar_outputs=True)
9846 @torch._dynamo.config.patch(capture_scalar_outputs=True)
[all …]
H A Dtest_decorators.py577 torch._dynamo.config.capture_scalar_outputs = True
/aosp_15_r20/external/pytorch/benchmarks/dynamo/pr_time_benchmarks/benchmarks/
H A Dupdate_hint_benchmark.py19 torch._dynamo.config.capture_scalar_outputs = True
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dconfig.py222 capture_scalar_outputs = os.environ.get("TORCHDYNAMO_CAPTURE_SCALAR_OUTPUTS") == "1" variable
H A Deval_frame.py1417 capture_scalar_outputs=True,
H A Doutput_graph.py298 allow_scalar_outputs=config.capture_scalar_outputs,
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_dynamo_distributed.py356 @config.patch(optimize_ddp=True, capture_scalar_outputs=True)
377 @config.patch(optimize_ddp=True, capture_scalar_outputs=True)
399 @config.patch(optimize_ddp=True, capture_scalar_outputs=True)
H A Dtest_c10d_functional_native.py801 capture_scalar_outputs=True,
H A Dtest_inductor_collectives.py430 capture_scalar_outputs=True,
/aosp_15_r20/external/pytorch/test/higher_order_ops/
H A Dtest_with_effects.py307 torch._dynamo.config.capture_scalar_outputs = True
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dtensor.py767 if not config.capture_scalar_outputs:
/aosp_15_r20/external/pytorch/test/export/
H A Dtest_converter.py867 torch._dynamo.config.capture_scalar_outputs = True
H A Dtest_export.py752 @torch._dynamo.config.patch(capture_scalar_outputs=True)
/aosp_15_r20/external/pytorch/benchmarks/dynamo/
H A Dcommon.py3706 torch._dynamo.config.capture_scalar_outputs = True