#0: skip N300 tests for ubench CI

tenstorrent · Feb 25, 2025 · e7182a0 · e7182a0
1 parent a32c401
commit e7182a0
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/tt_metal/microbenchmarks/ethernet/test_fabric_edm_bandwidth.py b/tests/tt_metal/microbenchmarks/ethernet/test_fabric_edm_bandwidth.py
@@ -93,8 +93,10 @@ def run_fabric_edm(
     [5.65],
 )
 def test_fabric_edm_mcast_bw(
-    num_mcasts, num_unicasts, num_links, num_op_invocations, line_sync, line_size, packet_size, expected_bw
+    mesh_device, num_mcasts, num_unicasts, num_links, num_op_invocations, line_sync, line_size, packet_size, expected_bw
 ):
+    if len(mesh_device.get_device_ids()) != 8:
+        pytest.skip("Skipping test because it only works with T3000")
     run_fabric_edm(
         False,
         num_mcasts,
@@ -120,8 +122,10 @@ def test_fabric_edm_mcast_bw(
     [7.13],
 )
 def test_fabric_edm_unicast_bw(
-    num_mcasts, num_unicasts, num_links, num_op_invocations, line_sync, line_size, packet_size, expected_bw
+    mesh_device, num_mcasts, num_unicasts, num_links, num_op_invocations, line_sync, line_size, packet_size, expected_bw
 ):
+    if len(mesh_device.get_device_ids()) != 8:
+        pytest.skip("Skipping test because it only works with T3000")
     run_fabric_edm(
         True,
         num_mcasts,