From 57c79cc98715a83097d0eb2617753644ff104cbb Mon Sep 17 00:00:00 2001 From: Aswinmcw Date: Fri, 25 Oct 2024 04:08:21 +0000 Subject: [PATCH] #5560: Use static methods --- .../experimental/ccl/all_reduce/device/all_reduce_op.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ttnn/cpp/ttnn/operations/experimental/ccl/all_reduce/device/all_reduce_op.cpp b/ttnn/cpp/ttnn/operations/experimental/ccl/all_reduce/device/all_reduce_op.cpp index dbaddc6f814..48c1a388e5d 100644 --- a/ttnn/cpp/ttnn/operations/experimental/ccl/all_reduce/device/all_reduce_op.cpp +++ b/ttnn/cpp/ttnn/operations/experimental/ccl/all_reduce/device/all_reduce_op.cpp @@ -61,7 +61,7 @@ namespace operations{ namespace experimental{ namespace ccl{ -AllReduceStrategy choose_all_reduce_strategy(const Tensor& input_tensor, uint32_t num_devices, uint32_t num_links) { +static AllReduceStrategy choose_all_reduce_strategy(const Tensor& input_tensor, uint32_t num_devices, uint32_t num_links) { auto shape = input_tensor.get_logical_shape(); auto rank = shape.rank(); @@ -98,7 +98,7 @@ AllReduceStrategy choose_all_reduce_strategy(const Tensor& input_tensor, uint32_ } -Tensor all_gather_local_reduce(const Tensor& input_tensor, uint32_t num_devices, uint32_t num_links, const MemoryConfig& output_mem_config, +static Tensor all_gather_local_reduce(const Tensor& input_tensor, uint32_t num_devices, uint32_t num_links, const MemoryConfig& output_mem_config, const std::optional user_defined_num_workers, const std::optional user_defined_num_buffers_per_channel, const std::vector& devices, const ttnn::ccl::Topology& topology) { auto shape = input_tensor.get_logical_shape(); @@ -125,7 +125,7 @@ Tensor all_gather_local_reduce(const Tensor& input_tensor, uint32_t num_devices, return ttnn::reshape(sum_tensor, shape); } -Tensor reduce_scatter_all_gather(const Tensor& input_tensor, const ttnn::operations::binary::BinaryOpType binary_op_type, uint32_t num_devices, uint32_t num_links, const MemoryConfig& output_mem_config, +static Tensor reduce_scatter_all_gather(const Tensor& input_tensor, const ttnn::operations::binary::BinaryOpType binary_op_type, uint32_t num_devices, uint32_t num_links, const MemoryConfig& output_mem_config, const std::optional user_defined_num_workers, const std::optional user_defined_num_buffers_per_channel, const std::vector& devices, const ttnn::ccl::Topology& topology) { auto shape = input_tensor.get_logical_shape(); auto rank = shape.rank();