Home
last modified time | relevance | path

Searched refs:run_and_get_cpp_code (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_memory_planning.py23 from torch._inductor.utils import run_and_get_cpp_code
52 result, code = run_and_get_cpp_code(compiled, *args)
69 result, code = run_and_get_cpp_code(compiled, *args)
93 result, code = run_and_get_cpp_code(
H A Dtest_extension_backend.py46 run_and_get_cpp_code = test_torchinductor.run_and_get_cpp_code variable
148 _, code = run_and_get_cpp_code(opt_fn, x, y, z)
H A Dtest_torchinductor_codegen_dynamic_shapes.py26 run_and_get_cpp_code,
78 _, code = run_and_get_cpp_code(run, *example_inputs, **kwargs)
H A Dtest_cpu_repro.py63 run_and_get_cpp_code = test_torchinductor.run_and_get_cpp_code variable
506 _, code = run_and_get_cpp_code(fn_opt, *inps)
600 _, code = run_and_get_cpp_code(fn_opt, *inps)
1916 _, code = run_and_get_cpp_code(_fn_opt, *inps)
2098 _, code = run_and_get_cpp_code(f_opt, inps[0], inps[1])
2111 _, code = run_and_get_cpp_code(f_opt, inps[0])
2854 _, code = run_and_get_cpp_code(opt_fn, x)
3039 _, code = run_and_get_cpp_code(run, v)
3127 _, code = run_and_get_cpp_code(fn_opt, *inps)
3394 _, code = run_and_get_cpp_code(m_opt, idx, x)
[all …]
H A Dtest_cuda_cpp_wrapper.py113 _, code = test_torchinductor.run_and_get_cpp_code(
H A Dtest_cpu_cpp_wrapper.py144 _, code = test_torchinductor.run_and_get_cpp_code(
H A Dtest_aot_inductor.py23 from torch._inductor.utils import run_and_get_cpp_code
3324 result, code = run_and_get_cpp_code(
3351 result, code = run_and_get_cpp_code(
3399 result, code = run_and_get_cpp_code(
H A Dtest_torchinductor.py51 run_and_get_cpp_code,
1362 _, code = run_and_get_cpp_code(fn_opt, *inps)
9451 _, code = run_and_get_cpp_code(opt_fn, *args)
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dutils.py1924 def run_and_get_cpp_code(fn, *args, **kwargs): function