Searched refs:test_triton_kernel (Results 1 – 2 of 2) sorted by relevance
/aosp_15_r20/external/pytorch/test/inductor/ | ||
H A D | test_aot_inductor.py | 1976 def test_triton_kernel(self, grid_type, num_dims, dynamic, autotune): member in AOTInductorTestsTemplate |
/aosp_15_r20/external/pytorch/test/ | ||
H A D | test_sparse_csr.py | 4031 def test_triton_kernel(self, op, device, dtype, blocksize): member in TestSparseCompressedTritonKernels |