#0: Seperate Config for Blackhole

tenstorrent · Nov 19, 2024 · f6281d2 · f6281d2
1 parent e96f47b
commit f6281d2
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/ttnn/ttnn/device.py b/ttnn/ttnn/device.py
@@ -157,14 +157,22 @@ def GetComputeKernelConfig(
     dst_full_sync_en=False,
 ):
     device = ttnn.GetDefaultDevice()
-    if is_wormhole_b0(device) or is_blackhole(device):
+    if is_wormhole_b0(device):
         return ttnn.WormholeComputeKernelConfig(
             math_fidelity=math_fidelity,
             fp32_dest_acc_en=fp32_dest_acc_en,
             packer_l1_acc=packer_l1_acc,
             math_approx_mode=math_approx_mode,
             dst_full_sync_en=dst_full_sync_en,
         )
+    elif is_blackhole(device):
+        return ttnn.BlackholeComputeKernelConfig(
+            math_fidelity=math_fidelity,
+            fp32_dest_acc_en=fp32_dest_acc_en,
+            packer_l1_acc=packer_l1_acc,
+            math_approx_mode=math_approx_mode,
+            dst_full_sync_en=dst_full_sync_en,
+        )
     else:
         return ttnn.GrayskullComputeKernelConfig(
             math_fidelity=math_fidelity, math_approx_mode=math_approx_mode, dst_full_sync_en=dst_full_sync_en