nimlgen/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Activity

rename amd/test_custom_kernel.py to test_asm_kernel (#16586)

Browse Source 
* rename amd/test_custom_kernel.py to test_asm_kernel

* update
This commit is contained in:
qazal
2026-06-12 16:11:01 +09:00
committed by GitHub
parent e770805d21
commit 12f4cf0e49
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extra/sqtt/examples/generate_examples.py
+2 -2
View File
@@ -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__":
test/amd/test_custom_kernel.py → test/amd/test_asm_kernel.py
+1 -1
View File
@@ -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):