IQ.Pilot Release Commit @ bec7652

artifacts/package_sources/tinygrad/extra/hcqfuzz/spec.py

@@ -0,0 +1,42 @@
+import os, random
+
+class TestSpec:
+  def prepare(self, device, seed):
+    raise NotImplementedError("prepare must be implemented in the derived class")
+  def get_exec_state(self):
+    raise NotImplementedError("get_exec_state must be implemented in the derived class")
+  def name(self): return self.__class__.__name__
+
+class DeviceSpec:
+  def prepare(self, seed):
+    raise NotImplementedError("prepare must be implemented in the derived class")
+  def get_exec_state(self):
+    raise NotImplementedError("get_exec_state must be implemented in the derived class")
+  def name(self): return self.__class__.__name__
+
+class HCQSpec(DeviceSpec): pass
+class AMDSpec(HCQSpec):
+  def __init__(self):
+    assert os.path.exists('/sys/module/amdgpu'), "amdgpu module should be loaded"
+
+  def prepare(self, seed):
+    self.env = {
+      "AMD": 1,
+      "AMD_LLVM": 0
+    }
+
+  def get_exec_state(self): return self.env
+
+class AMSpec(AMDSpec):
+  def __init__(self):
+    assert not os.path.exists('/sys/module/amdgpu'), "amdgpu module should not be loaded"
+
+  def prepare(self, seed):
+    super().prepare(seed)
+
+    self.env = {
+      **self.env, # from AMDSpec
+      "AMD_SDMA_BIND": random.randint(0, 1),
+      "AMD_ALLOC_QUEUE_DEV_MEM": 0, # random.randint(0, 1) need to validate
+      "AMD_QUEUE_SIZE": 1 << random.randint(10, 26),
+    }