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

tensor cores for gfx1200 [pr] (#10795)

Browse Source
This commit is contained in:
George Hotz
2025-06-12 16:33:29 -07:00
committed by GitHub
parent a113c5e3ae
commit dcd1928f29
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/renderer/cstyle.py
+3 -1
View File
@@ -422,7 +422,9 @@ class AMDRenderer(CStyleLanguage):
@staticmethod
def get_tensor_cores(arch):
return {"gfx942": AMDRenderer.tensor_cores_mfma, "gfx1201": AMDRenderer.tensor_cores_rdna4}.get(arch.split(":")[0], AMDRenderer.tensor_cores)
return {"gfx942": AMDRenderer.tensor_cores_mfma,
"gfx1200": AMDRenderer.tensor_cores_rdna4,
"gfx1201": AMDRenderer.tensor_cores_rdna4}.get(arch.split(":")[0], AMDRenderer.tensor_cores)
def __init__(self, arch:str): # gfx942 => MI300, gfx1100 => RX 7900, gfx1201 => RX 9700
self.arch = arch
self.tensor_cores = self.get_tensor_cores(arch)