#0: Rebased

tenstorrent · Jul 22, 2024 · 635950e · 635950e
1 parent a578d30
commit 635950e
Show file tree

Hide file tree

Showing 19 changed files with 31 additions and 1,906 deletions.
diff --git a/tests/tt_eager/ops/ccl/test_all_gather_sharded_indexing_helpers.cpp b/tests/tt_eager/ops/ccl/test_all_gather_sharded_indexing_helpers.cpp
@@ -289,4 +289,4 @@ TEST(AllGatherSharded_WidthShardedIndexing_FullWorkerGridVariant, AdvanceSingleT
             }
         }
     }
-}
+}
diff --git a/tests/tt_eager/ops/ccl/test_all_gather_utils.cpp b/tests/tt_eager/ops/ccl/test_all_gather_utils.cpp
@@ -1364,4 +1364,4 @@ TEST(AllGatherUtilsDevice, AddrGenAdvanceWidthSharded_RingSize8RingIndex0_NumWor
     ASSERT_EQ(curr_core_chunk_index, 4);
     ASSERT_EQ(curr_worker_index, 6);
     ASSERT_EQ(contiguous_chunk_count, 1);
-}
+}
diff --git a/ttnn/cpp/pybind11/operations/__init__.hpp b/ttnn/cpp/pybind11/operations/__init__.hpp
@@ -64,14 +64,10 @@ void py_module(py::module& module) {
     auto m_unary_backward = module.def_submodule("unary_backward", "unary_backward operations");
     unary_backward::py_module(m_unary_backward);
 
-    auto m_ccl = module.def_submodule("ccl", "collective communication operations");
-    ccl::py_module_all_gather(m_ccl);
-    ccl::py_module_line_all_gather(m_ccl);
-    ccl::py_module_reduce_scatter(m_ccl);
-
     auto m_ccl = module.def_submodule("ccl", "collective communication operations");
     ccl::py_bind_all_gather(m_ccl);
     ccl::py_bind_line_all_gather(m_ccl);
+    ccl::py_bind_reduce_scatter(m_ccl);
 
     auto m_complex_unary = module.def_submodule("complex_unary", "complex_unary operations");
     complex_unary::py_module(m_complex_unary);

diff --git a/...xperimental/tt_dnn/op_library/ccl/reduce_scatter/host/reduce_scatter_full_worker_grid.cpp b/...xperimental/tt_dnn/op_library/ccl/reduce_scatter/host/reduce_scatter_full_worker_grid.cpp
diff --git a/...n/op_library/ccl/reduce_scatter/kernels/worker_interleaved_ring_reduce_scatter_reader.cpp b/...n/op_library/ccl/reduce_scatter/kernels/worker_interleaved_ring_reduce_scatter_reader.cpp
diff --git a/...n/op_library/ccl/reduce_scatter/kernels/worker_interleaved_ring_reduce_scatter_sender.cpp b/...n/op_library/ccl/reduce_scatter/kernels/worker_interleaved_ring_reduce_scatter_sender.cpp
diff --git a/ttnn/cpp/ttnn/experimental/tt_dnn/op_library/ccl/reduce_scatter/reduce_scatter_op.cpp b/ttnn/cpp/ttnn/experimental/tt_dnn/op_library/ccl/reduce_scatter/reduce_scatter_op.cpp
diff --git a/ttnn/cpp/ttnn/experimental/tt_dnn/op_library/ccl/reduce_scatter/reduce_scatter_op.hpp b/ttnn/cpp/ttnn/experimental/tt_dnn/op_library/ccl/reduce_scatter/reduce_scatter_op.hpp
-Original file line number
+Diff line change
@@ Expand Up @@
                 }
             }
         }
-    }
+    }