Searched refs:run_and_get_triton_code (Results 1 – 10 of 10) sorted by relevance
/aosp_15_r20/external/pytorch/test/distributed/ |
H A D | test_c10d_functional_native.py | 10 from torch._inductor.utils import fresh_inductor_cache, run_and_get_triton_code 245 code = run_and_get_triton_code(compiled, arg) 454 code = run_and_get_triton_code(compiled, arg) 517 code = run_and_get_triton_code(compiled, arg) 553 code = run_and_get_triton_code(compiled, args) 596 code = run_and_get_triton_code(compiled, arg) 627 code = run_and_get_triton_code(compiled, arg) 656 code = run_and_get_triton_code(compiled, arg) 683 code = run_and_get_triton_code(compiled, args) 717 code = run_and_get_triton_code(compiled, arg) [all …]
|
H A D | test_compute_comm_reordering.py | 24 from torch._inductor.utils import run_and_get_triton_code 119 code = run_and_get_triton_code(compiled, inputs) 160 code = run_and_get_triton_code(compiled, inputs) 209 code = run_and_get_triton_code(compiled, inputs, **self.get_world_trs()) 262 code = run_and_get_triton_code(compiled, inputs, **self.get_world_trs()) 319 code = run_and_get_triton_code(compiled, inputs, **self.get_world_trs()) 368 code = run_and_get_triton_code(compiled, inputs, **self.get_world_trs())
|
H A D | test_inductor_collectives.py | 17 from torch._inductor.utils import run_and_get_triton_code 449 code = run_and_get_triton_code(compiled_fn, *inputs, **trs) 489 code = run_and_get_triton_code(compiled_fn, *inputs, **trs) 535 code = run_and_get_triton_code(compiled, inputs, **self.get_world_trs()) 569 code = run_and_get_triton_code(compiled, inputs, **self.get_world_trs()) 604 code = run_and_get_triton_code(compiled, inputs, **self.get_world_trs()) 1055 code = run_and_get_triton_code(compiled, inputs, **self.get_world_trs()) 1101 code = run_and_get_triton_code(compiled, inputs, **self.get_world_trs())
|
/aosp_15_r20/external/pytorch/test/distributed/tensor/parallel/ |
H A D | test_micro_pipeline_tp.py | 16 from torch._inductor.utils import fresh_inductor_cache, run_and_get_triton_code 223 code = run_and_get_triton_code(compiled, A_shard, B) 291 code = run_and_get_triton_code( 325 code = run_and_get_triton_code(compiled, A, B) 378 code = run_and_get_triton_code( 407 code = run_and_get_triton_code(compiled, inp)
|
/aosp_15_r20/external/pytorch/test/inductor/ |
H A D | test_torchinductor_codegen_dynamic_shapes.py | 27 run_and_get_triton_code, 81 code = run_and_get_triton_code(run, *example_inputs, **kwargs)
|
H A D | test_indexing.py | 15 from torch._inductor.utils import run_and_get_triton_code 236 triton_code = run_and_get_triton_code(f, x)
|
H A D | test_torchinductor.py | 52 run_and_get_triton_code, 1368 code = run_and_get_triton_code(fn_opt, *inps) 7508 code = run_and_get_triton_code(torch.compile(f), torch.zeros((4, 2))) 10690 code = run_and_get_triton_code(fn_opt, *inps) 10716 code1 = run_and_get_triton_code(fn1_opt, *inps) 10717 code2 = run_and_get_triton_code(fn2_opt, *inps) 10827 code = run_and_get_triton_code(fn_opt, x) 10853 code = run_and_get_triton_code(fn_opt, *inps) 10876 code = run_and_get_triton_code(fn_opt, *inps) 10896 code = run_and_get_triton_code(fn_gpu, x) [all …]
|
/aosp_15_r20/external/pytorch/test/distributed/_composable/ |
H A D | test_replicate_with_compiler.py | 17 from torch._inductor.utils import run_and_get_triton_code 295 code = run_and_get_triton_code(functools.partial(bwd, loss=loss))
|
/aosp_15_r20/external/pytorch/test/distributed/_tensor/ |
H A D | test_dtensor_compile.py | 15 from torch._inductor.utils import run_and_get_triton_code 715 code = run_and_get_triton_code(compiled_model, inp)
|
/aosp_15_r20/external/pytorch/torch/_inductor/ |
H A D | utils.py | 1359 def run_and_get_triton_code(fn, *args, **kwargs): function
|