Home
last modified time | relevance | path

Searched refs:autotune_select_algorithm (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/kernel/
H A Dmm.py27 autotune_select_algorithm,
249 return autotune_select_algorithm(name, choices, [mat1, mat2], layout)
313 return autotune_select_algorithm("int_mm", choices, [mat1, mat2], layout)
319 return autotune_select_algorithm("int_mm", choices, [mat1, mat2], layout)
348 return autotune_select_algorithm("addmm", choices, [inp, mat1, mat2], layout)
452 return autotune_select_algorithm(
509 return autotune_select_algorithm(
750 return autotune_select_algorithm(name, choices, input_nodes, layout)
776 return autotune_select_algorithm("int_mm", choices, [mat1, mat2, mat3], layout)
H A Dbmm.py8 autotune_select_algorithm,
167 return autotune_select_algorithm("bmm", choices, [mat1, mat2], layout)
192 return autotune_select_algorithm("baddbmm", choices, [inp, mat1, mat2], layout)
H A Dunpack_mixed_mm.py5 from ..select_algorithm import autotune_select_algorithm, TritonTemplate
87 return autotune_select_algorithm("uint4x2_mixed_mm", choices, [mat1, mat2], layout)
H A Dmm_plus_mm.py8 autotune_select_algorithm,
246 return autotune_select_algorithm(
H A Dmm_scaled.py12 autotune_select_algorithm,
304 return autotune_select_algorithm("scaled_mm", choices, input_nodes, layout)
H A Dflex_attention.py29 from ..select_algorithm import autotune_select_algorithm, realize_inputs, TritonTemplate
895 autotune_select_algorithm(
1832 grad_key = autotune_select_algorithm(
H A Dflex_decoding.py14 from ..select_algorithm import autotune_select_algorithm, TritonTemplate
535 buf_ACC = autotune_select_algorithm(
H A Dconv.py19 autotune_select_algorithm,
647 return autotune_select_algorithm("convolution", choices, args, layout)
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dquantized_lowerings.py12 autotune_select_algorithm,
90 return autotune_select_algorithm(
H A Dmkldnn_lowerings.py24 autotune_select_algorithm,
220 result = autotune_select_algorithm(
282 result = autotune_select_algorithm(
709 result = autotune_select_algorithm(
1012 result = autotune_select_algorithm(
1074 result: TensorBox = autotune_select_algorithm(
H A Dselect_algorithm.py1714 def autotune_select_algorithm(*args, **kwargs): function
/aosp_15_r20/external/pytorch/torchgen/_autoheuristic/
H A DREADME.md41 This call has to be followed by a call to `autotune_select_algorithm()`,
43 autotune_select_algorithm(name, choices, input_nodes, layout)
45 …nodes`, and `name` in the `AutoHeuristicSelectAlgorithm()` and `autotune_select_algorithm()` calls…
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Db2b_gemm.py28 autotune_select_algorithm,
556 return autotune_select_algorithm("b2b_gemm", choices, [A, B, C], layout)