Skip to content

Commit

Permalink
more namespace missing
Browse files Browse the repository at this point in the history
  • Loading branch information
kpaigwar committed Mar 7, 2025
1 parent 1201038 commit 400c7e5
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -165,7 +165,7 @@ const tt::tt_metal::operation::Hash AllReduceAsync::compute_program_hash(
auto input_dtype = input_tensors[0].get_dtype();
auto input_memory_config = input_tensors[0].memory_config();

return operation::hash_operation<AllReduceAsync>(
return tt::tt_metal::operation::hash_operation<AllReduceAsync>(
this->num_links,
this->ring_size,
this->ring_index,
Expand Down Expand Up @@ -199,10 +199,10 @@ Tensor all_reduce_async(
auto devices = input_tensor.get_workers();
std::size_t num_devices = (cluster_axis == 0) ? mesh_view.num_rows() : mesh_view.num_cols();

std::vector<Tensor> output_tensors = {Tensor(operation::get_workers_for_op_output({input_tensor}))};
std::vector<Tensor> output_tensors = {Tensor(tt::tt_metal::operation::get_workers_for_op_output({input_tensor}))};
std::vector<GlobalSemaphore> semaphores = multi_device_global_semaphore.global_semaphores;

operation::launch_op(
tt::tt_metal::operation::launch_op(
[num_preferred_links,
memory_config,
mesh_view,
Expand All @@ -227,7 +227,7 @@ Tensor all_reduce_async(
const auto& input_tensor = input_tensors.at(0);
const auto& buffer_tensor = input_tensors.at(1);

return operation::run(
return tt::tt_metal::operation::run(
ttnn::ccl::all_reduce_detail::create_all_reduce_async_struct(
input_device_tensor,
num_preferred_links.has_value() ? num_preferred_links.value() : 1,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -298,7 +298,9 @@ tt::tt_metal::operation::ProgramWithCallbacks all_reduce_async_minimal_multi_cor
"worker_reader.cpp",
sender_worker_core_range,
tt::tt_metal::DataMovementConfig{
.processor = DataMovementProcessor::RISCV_1, .noc = reader_noc, .compile_args = reader_compile_args});
.processor = tt::tt_metal::DataMovementProcessor::RISCV_1,
.noc = reader_noc,
.compile_args = reader_compile_args});

// Writer
std::vector<uint32_t> writer_compile_args = {
Expand All @@ -318,7 +320,9 @@ tt::tt_metal::operation::ProgramWithCallbacks all_reduce_async_minimal_multi_cor
"worker_writer.cpp",
sender_worker_core_range,
tt::tt_metal::DataMovementConfig{
.processor = DataMovementProcessor::RISCV_0, .noc = writer_noc, .compile_args = writer_compile_args});
.processor = tt::tt_metal::DataMovementProcessor::RISCV_0,
.noc = writer_noc,
.compile_args = writer_compile_args});

// Kernel Runtime Args
for (uint32_t link = 0; link < num_links; link++) {
Expand Down

0 comments on commit 400c7e5

Please sign in to comment.