#17080: Add blackhole valid ranges

tenstorrent · Jan 24, 2025 · d63cfb2 · d63cfb2
1 parent 3fa1761
commit d63cfb2
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/tests/tt_metal/tools/profiler/test_device_profiler.py b/tests/tt_metal/tools/profiler/test_device_profiler.py
@@ -21,7 +21,7 @@
     clear_profiler_runtime_artifacts,
 )
 
-from models.utility_functions import skip_for_grayskull
+from models.utility_functions import skip_for_grayskull, skip_for_blackhole
 
 PROG_EXMP_DIR = "programming_examples/profiler"
 
@@ -82,6 +82,7 @@ def test_multi_op():
     REF_COUNT_DICT = {
         "grayskull": [108 * OP_COUNT * RUN_COUNT, 88 * OP_COUNT * RUN_COUNT],
         "wormhole_b0": [72 * OP_COUNT * RUN_COUNT, 64 * OP_COUNT * RUN_COUNT, 56 * OP_COUNT * RUN_COUNT],
+        "blackhole": [72 * OP_COUNT * RUN_COUNT, 64 * OP_COUNT * RUN_COUNT, 56 * OP_COUNT * RUN_COUNT],
     }
 
     ENV_VAR_ARCH_NAME = os.getenv("ARCH_NAME")
@@ -152,6 +153,11 @@ def test_full_buffer():
             64 * OP_COUNT * RISC_COUNT * ZONE_COUNT,
             56 * OP_COUNT * RISC_COUNT * ZONE_COUNT,
         ],
+        "blackhole": [
+            72 * OP_COUNT * RISC_COUNT * ZONE_COUNT,
+            64 * OP_COUNT * RISC_COUNT * ZONE_COUNT,
+            56 * OP_COUNT * RISC_COUNT * ZONE_COUNT,
+        ],
     }
 
     ENV_VAR_ARCH_NAME = os.getenv("ARCH_NAME")
@@ -189,6 +195,10 @@ def test_dispatch_cores():
             "Tensix CQ Dispatch": 16,
             "Tensix CQ Prefetch": 25,
         },
+        "blackhole": {
+            "Tensix CQ Dispatch": 16,
+            "Tensix CQ Prefetch": 25,
+        },
     }
 
     ENV_VAR_ARCH_NAME = os.getenv("ARCH_NAME")
@@ -216,6 +226,7 @@ def test_dispatch_cores():
     os.environ["TT_METAL_DEVICE_PROFILER_DISPATCH"] = "0"
 
 
+@skip_for_blackhole()
 @skip_for_grayskull()
 def test_ethernet_dispatch_cores():
     REF_COUNT_DICT = {
@@ -308,9 +319,11 @@ def test_timestamped_events():
     REF_COUNT_DICT = {
         "grayskull": [108 * OP_COUNT * RISC_COUNT * ZONE_COUNT, 88 * OP_COUNT * RISC_COUNT * ZONE_COUNT],
         "wormhole_b0": [(T * RISC_COUNT + E) * OP_COUNT * ZONE_COUNT for T, E in COMBO_COUNTS],
+        "blackhole": [(T * RISC_COUNT + E) * OP_COUNT * ZONE_COUNT for T, E in COMBO_COUNTS],
     }
     REF_ERISC_COUNT = {
         "wormhole_b0": [C * OP_COUNT * ZONE_COUNT for C in ERISC_COUNTS],
+        "blackhole": [C * OP_COUNT * ZONE_COUNT for C in ERISC_COUNTS],
     }
 
     ENV_VAR_ARCH_NAME = os.getenv("ARCH_NAME")