Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dutils.py1069 def _use_autotune_backend(backend: str) -> bool: function
1091 and _use_autotune_backend("TRITON")
1109 res = _use_template_for_cuda(layout, layout_dtypes) and _use_autotune_backend(
1162 if not _use_autotune_backend("CK"):
1221 if not _use_template_for_cpu(layout) or not _use_autotune_backend("CPP"):
1268 return not use_max_autotune() or _use_autotune_backend("ATEN")