Searched refs:capture_scalar_outputs (Results 1 – 23 of 23) sorted by relevance
/aosp_15_r20/external/pytorch/test/inductor/ |
H A D | test_torchinductor_dynamic_shapes.py | 243 @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 D | test_unbacked_symints.py | 264 @torch._dynamo.config.patch(capture_scalar_outputs=True)
|
H A D | test_auto_functionalize.py | 340 capture_scalar_outputs=True, capture_dynamic_output_shape_ops=True
|
H A D | test_torchinductor.py | 4906 @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 D | test_triton_kernels.py | 1115 @torch._dynamo.config.patch(capture_scalar_outputs=True)
|
H A D | test_aot_inductor.py | 2786 @torch._dynamo.config.patch(capture_scalar_outputs=True)
|
/aosp_15_r20/external/pytorch/test/dynamo/ |
H A D | test_unspec.py | 456 @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 D | test_export.py | 468 @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 D | test_repros.py | 1347 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 D | test_misc.py | 1377 @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 D | test_decorators.py | 577 torch._dynamo.config.capture_scalar_outputs = True
|
/aosp_15_r20/external/pytorch/benchmarks/dynamo/pr_time_benchmarks/benchmarks/ |
H A D | update_hint_benchmark.py | 19 torch._dynamo.config.capture_scalar_outputs = True
|
/aosp_15_r20/external/pytorch/torch/_dynamo/ |
H A D | config.py | 222 capture_scalar_outputs = os.environ.get("TORCHDYNAMO_CAPTURE_SCALAR_OUTPUTS") == "1" variable
|
H A D | eval_frame.py | 1417 capture_scalar_outputs=True,
|
H A D | output_graph.py | 298 allow_scalar_outputs=config.capture_scalar_outputs,
|
/aosp_15_r20/external/pytorch/test/distributed/ |
H A D | test_dynamo_distributed.py | 356 @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 D | test_c10d_functional_native.py | 801 capture_scalar_outputs=True,
|
H A D | test_inductor_collectives.py | 430 capture_scalar_outputs=True,
|
/aosp_15_r20/external/pytorch/test/higher_order_ops/ |
H A D | test_with_effects.py | 307 torch._dynamo.config.capture_scalar_outputs = True
|
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/ |
H A D | tensor.py | 767 if not config.capture_scalar_outputs:
|
/aosp_15_r20/external/pytorch/test/export/ |
H A D | test_converter.py | 867 torch._dynamo.config.capture_scalar_outputs = True
|
H A D | test_export.py | 752 @torch._dynamo.config.patch(capture_scalar_outputs=True)
|
/aosp_15_r20/external/pytorch/benchmarks/dynamo/ |
H A D | common.py | 3706 torch._dynamo.config.capture_scalar_outputs = True
|