From 31bcbed6bba7f5ae2b552b372f91af29984aad22 Mon Sep 17 00:00:00 2001
From: George Hotz <72895+geohot@users.noreply.github.com>
Date: Mon, 19 Jan 2026 16:10:30 +0900
Subject: [PATCH] AMD_DISABLE_SDMA for testing with -n12 (#14216)

---
 test/test_jit.py            | 1 +
 tinygrad/runtime/ops_amd.py | 1 +
 2 files changed, 2 insertions(+)

diff --git a/test/test_jit.py b/test/test_jit.py
index c53651d5d1..2025091432 100644
--- a/test/test_jit.py
+++ b/test/test_jit.py
@@ -806,6 +806,7 @@ class TestJitGraphSplit(unittest.TestCase):
       multigraph=[self.ji_graph(6)],
       hcqgraph=[self.ji_graph(6)])
 
+  @unittest.skip("this fails if you don't have SDMA or are using AMD_DISABLE_SDMA=1")
   @unittest.skipIf(getenv("MOCKGPU"), "MockGPU does not support parallel copies")
   def test_jit_multidev_copy(self):
     if Device.DEFAULT in {"CPU"}: raise unittest.SkipTest("CPU/LLVM is not a valid default device for this test (zero-copies)")
diff --git a/tinygrad/runtime/ops_amd.py b/tinygrad/runtime/ops_amd.py
index 06d06b8953..8d3be665dd 100644
--- a/tinygrad/runtime/ops_amd.py
+++ b/tinygrad/runtime/ops_amd.py
@@ -1007,6 +1007,7 @@ class AMDDevice(HCQCompiled):
 
   @functools.lru_cache(None)
   def sdma_queue(self, idx:int):
+    if getenv("AMD_DISABLE_SDMA"): return None
     with contextlib.suppress(OSError): return self.create_queue(kfd.KFD_IOC_QUEUE_TYPE_SDMA, 0x200 if self.is_usb() else (16 << 20), idx=idx)
     return None