#0: Rebased

tenstorrent · Oct 1, 2024 · ea0e2f4 · ea0e2f4
1 parent c1f11a4
commit ea0e2f4
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/ttnn/unit_tests/operations/test_all_gather.py b/tests/ttnn/unit_tests/operations/test_all_gather.py
@@ -212,7 +212,7 @@ def run_all_gather_on_n300_impl(
         mem_config,
         use_program_cache,
         function_level_defaults,
-        all_gather_topology,
+        all_gather_topology=all_gather_topology,
         num_iters=num_iters,
         enable_async=enable_async,
     )
@@ -253,7 +253,7 @@ def run_all_gather_on_t3000_impl(
         mem_config,
         use_program_cache,
         function_level_defaults,
-        all_gather_topology,
+        all_gather_topology=all_gather_topology,
         num_iters=num_iters,
         enable_async=enable_async,
     )
@@ -285,7 +285,7 @@ def run_all_gather_on_t3000_impl_tight_loop(
         mem_config,
         use_program_cache,
         function_level_defaults,
-        all_gather_topology,
+        all_gather_topology=all_gather_topology,
         num_iters=num_iters,
         enable_async=enable_async,
     )
@@ -463,7 +463,7 @@ def test_all_gather_on_t3000_nightly_commit_looping_4chip_ring(
         mem_config,
         use_program_cache,
         function_level_defaults,
-        all_gather_operation=ttnn.all_gather,
+        all_gather_topology=ttnn.Topology.Ring,
         num_iters=num_iters,
         enable_async=enable_async,
     )
@@ -512,7 +512,7 @@ def test_all_gather_on_t3000_post_commit_for_profiler_regression(
         mem_config,
         use_program_cache,
         function_level_defaults,
-        all_gather_operation=ttnn.all_gather,
+        all_gather_topology=ttnn.Topology.Ring,
     )