diff --git a/extra/sqtt/examples/generate_examples.py b/extra/sqtt/examples/generate_examples.py index 55ea70e4c4..a7014fbccb 100644 --- a/extra/sqtt/examples/generate_examples.py +++ b/extra/sqtt/examples/generate_examples.py @@ -9,8 +9,8 @@ EXAMPLES = { "empty":"test/backend/test_custom_kernel.py TestCustomKernel.test_empty", "plus":"test/test_tiny.py TestTiny.test_plus", "gemm":"-c \"from tinygrad import Tensor; (Tensor.empty(N:=32, N)@Tensor.empty(N, N)).realize()\"", - "sync":"test/amd/test_custom_kernel.py TestCustomKernel.test_lds_sync", - "handwritten":"test/amd/test_custom_kernel.py TestCustomKernel.test_handwritten", + "sync":"test/amd/test_asm_kernel.py TestAsmKernel.test_lds_sync", + "handwritten":"test/amd/test_asm_kernel.py TestAsmKernel.test_handwritten", } if __name__ == "__main__": diff --git a/test/amd/test_custom_kernel.py b/test/amd/test_asm_kernel.py similarity index 99% rename from test/amd/test_custom_kernel.py rename to test/amd/test_asm_kernel.py index 0d636c66fb..ebf269f1ca 100644 --- a/test/amd/test_custom_kernel.py +++ b/test/amd/test_asm_kernel.py @@ -162,7 +162,7 @@ def custom_data_deps(A:UOp) -> UOp: return UOp(Ops.PROGRAM, src=(sink, UOp(Ops.DEVICE, arg="AMD"), UOp(Ops.LINEAR, src=tuple([UOp(Ops.INS, arg=x) for x in insts])))) @unittest.skipUnless(Device.DEFAULT == "AMD", "requires AMD device") -class TestCustomKernel(unittest.TestCase): +class TestAsmKernel(unittest.TestCase): def setUp(self): self.arch = TARGET_TO_ARCH[Device["AMD"].arch] def test_simple(self):