Searched defs:max_autotune_gemm_backends (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_ck_backend.py | 67 self, max_autotune_gemm_backends, autotune_in_subproc argument
105 self, max_autotune_gemm_backends, autotune_in_subproc argument
150 def test_max_autotune_precompile_preselected(self, max_autotune_gemm_backends): argument
185 def test_max_autotune_precompile_non_contiguous(self, max_autotune_gemm_backends): argument
223 def test_max_autotune_addmm(self, max_autotune_gemm_backends, x_shape): argument
|
| H A D | test_cutlass_backend.py | 451 self, dynamic, max_autotune_gemm_backends argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | config.py | 301 max_autotune_gemm_backends = os.environ.get( variable
|