Home
last modified time | relevance | path

Searched refs:run_and_get_triton_code (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_c10d_functional_native.py10 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 Dtest_compute_comm_reordering.py24 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 Dtest_inductor_collectives.py17 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 Dtest_micro_pipeline_tp.py16 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 Dtest_torchinductor_codegen_dynamic_shapes.py27 run_and_get_triton_code,
81 code = run_and_get_triton_code(run, *example_inputs, **kwargs)
H A Dtest_indexing.py15 from torch._inductor.utils import run_and_get_triton_code
236 triton_code = run_and_get_triton_code(f, x)
H A Dtest_torchinductor.py52 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 Dtest_replicate_with_compiler.py17 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 Dtest_dtensor_compile.py15 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 Dutils.py1359 def run_and_get_triton_code(fn, *args, **kwargs): function