Searched refs:force_multi_kernel (Results 1 – 1 of 1) sorted by relevance
238 def test_reduction_scratch_buffer(self, force_multi_kernel=1): argument259 with config.patch("triton.multi_kernel", force_multi_kernel):263 def test_split_scan(self, force_multi_kernel=1): argument270 with config.patch("triton.multi_kernel", force_multi_kernel):274 def test_sort_disables_multi_kernel(self, force_multi_kernel=1): argument285 with config.patch("triton.multi_kernel", force_multi_kernel):291 test_reduction_scratch_buffer, force_multi_kernel=1297 make_cpp_wrapper_test(test_reduction_scratch_buffer, force_multi_kernel=2)301 make_cpp_wrapper_test(test_reduction_scratch_buffer, force_multi_kernel=3)