Fix output shape

tenstorrent · Jan 29, 2025 · 7e1705f · 7e1705f
1 parent 493a4f3
commit 7e1705f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/ttnn/unit_tests/operations/ccl/perf/test_ccl_async_perf.py b/tests/ttnn/unit_tests/operations/ccl/perf/test_ccl_async_perf.py
@@ -59,6 +59,7 @@ def test_all_gather_async_t3000(
     function_level_defaults,
     enable_async,
 ):
+    output_shape[dim] *= num_devices
     run_all_gather_impl(
         t3k_mesh_device,
         num_devices,