diff --git a/tinygrad/renderer/cstyle.py b/tinygrad/renderer/cstyle.py index 0d6bb11295..15592b8ece 100644 --- a/tinygrad/renderer/cstyle.py +++ b/tinygrad/renderer/cstyle.py @@ -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)