From ea0e2f41658310ba20ba36a3b8a298877c3d3ec2 Mon Sep 17 00:00:00 2001 From: Aswinmcw Date: Tue, 1 Oct 2024 04:22:02 +0000 Subject: [PATCH] #0: Rebased --- tests/ttnn/unit_tests/operations/test_all_gather.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/ttnn/unit_tests/operations/test_all_gather.py b/tests/ttnn/unit_tests/operations/test_all_gather.py index ba26acefa31..1e27f02ecb5 100644 --- 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, )