updated ccl perf target for llama all gather async

tenstorrent · Feb 10, 2025 · ccd3f98 · ccd3f98
1 parent 503ca1c
commit ccd3f98
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/ttnn/unit_tests/operations/ccl/test_ccl_async_TG_llama.py b/tests/ttnn/unit_tests/operations/ccl/test_ccl_async_TG_llama.py
@@ -112,7 +112,7 @@ def get_core_range_set(output_core_grid):
                 }
             ),
             ttnn.TILE_LAYOUT,
-            40,
+            32,
         ),
         (  # AllGather after Binary Mult+Silu
             ttnn.TensorMemoryLayout.WIDTH_SHARDED,
@@ -123,7 +123,7 @@ def get_core_range_set(output_core_grid):
             (32, 160),
             get_core_range_set(PREFETCHER_NOC1_RING),
             ttnn.TILE_LAYOUT,
-            32,
+            25,
         ),
     ),
 )