Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_aot_inductor.py1976 def test_triton_kernel(self, grid_type, num_dims, dynamic, autotune): member in AOTInductorTestsTemplate
/aosp_15_r20/external/pytorch/test/
H A Dtest_sparse_csr.py4031 def test_triton_kernel(self, op, device, dtype, blocksize): member in TestSparseCompressedTritonKernels