From 4a2bb011ab6f050e7f5f52c0f5391a1b46097ea6 Mon Sep 17 00:00:00 2001 From: Aswinmcw Date: Mon, 9 Dec 2024 10:14:18 +0000 Subject: [PATCH] Add CMakeLists for experimental CCL ops --- ttnn/CMakeLists.txt | 4 +++- ttnn/cpp/ttnn/operations/ccl/CMakeLists.txt | 8 -------- .../ttnn/operations/experimental/ccl/CMakeLists.txt | 12 ++++++++++++ 3 files changed, 15 insertions(+), 9 deletions(-) create mode 100644 ttnn/cpp/ttnn/operations/experimental/ccl/CMakeLists.txt diff --git a/ttnn/CMakeLists.txt b/ttnn/CMakeLists.txt index 66bab5cacbf..26c65848222 100644 --- a/ttnn/CMakeLists.txt +++ b/ttnn/CMakeLists.txt @@ -586,12 +586,14 @@ endforeach() ### Setup TTNN as a shared library with optional Python bindings add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/cpp/ttnn/tensor) add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/cpp/ttnn/operations/ccl) +add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/cpp/ttnn/operations//experimental/ccl) add_subdirectory(cpp/ttnn/deprecated) set(TTNN_FINAL_SRC ${TTNN_SRC} - ${CCL_TTNN_SRCS} ${QUEUE_SRCS} ${TENSOR_SRCS} + ${CCL_TTNN_SRCS} + ${CCL_EXPERIMENTAL_TTNN_SRCS} ${TT_DNN_SRCS} ) diff --git a/ttnn/cpp/ttnn/operations/ccl/CMakeLists.txt b/ttnn/cpp/ttnn/operations/ccl/CMakeLists.txt index 65fe305c215..148d928be91 100644 --- a/ttnn/cpp/ttnn/operations/ccl/CMakeLists.txt +++ b/ttnn/cpp/ttnn/operations/ccl/CMakeLists.txt @@ -21,14 +21,6 @@ set(CCL_TTNN_SRCS ${CMAKE_CURRENT_SOURCE_DIR}/barrier/device/barrier_op.cpp ${CMAKE_CURRENT_SOURCE_DIR}/barrier/barrier.cpp ${CMAKE_CURRENT_SOURCE_DIR}/barrier/barrier_pybind.cpp - #Experimental - ${CMAKE_CURRENT_SOURCE_DIR}/../experimental/ccl/all_gather_matmul/all_gather_matmul.cpp - ${CMAKE_CURRENT_SOURCE_DIR}/../experimental/ccl/all_gather_matmul/all_gather_matmul_pybind.cpp - ${CMAKE_CURRENT_SOURCE_DIR}/../experimental/ccl/all_gather_matmul/device/all_gather_matmul_op.cpp - ${CMAKE_CURRENT_SOURCE_DIR}/../experimental/ccl/all_gather_matmul/device/multi_core/all_gather_matmul_op_multi_core.cpp - ${CMAKE_CURRENT_SOURCE_DIR}/../experimental/ccl/all_reduce/all_reduce.cpp - ${CMAKE_CURRENT_SOURCE_DIR}/../experimental/ccl/all_reduce/all_reduce_pybind.cpp - ${CMAKE_CURRENT_SOURCE_DIR}/../experimental/ccl/all_reduce/device/all_reduce_op.cpp CACHE INTERNAL "CCL sources to reuse in ttnn build" ) diff --git a/ttnn/cpp/ttnn/operations/experimental/ccl/CMakeLists.txt b/ttnn/cpp/ttnn/operations/experimental/ccl/CMakeLists.txt new file mode 100644 index 00000000000..82767c44a09 --- /dev/null +++ b/ttnn/cpp/ttnn/operations/experimental/ccl/CMakeLists.txt @@ -0,0 +1,12 @@ +set(CCL_EXPERIMENTAL_TTNN_SRCS + #Experimental Ops + ${CMAKE_CURRENT_SOURCE_DIR}/all_gather_matmul/all_gather_matmul.cpp + ${CMAKE_CURRENT_SOURCE_DIR}/all_gather_matmul/all_gather_matmul_pybind.cpp + ${CMAKE_CURRENT_SOURCE_DIR}/all_gather_matmul/device/all_gather_matmul_op.cpp + ${CMAKE_CURRENT_SOURCE_DIR}/all_gather_matmul/device/multi_core/all_gather_matmul_op_multi_core.cpp + ${CMAKE_CURRENT_SOURCE_DIR}/all_reduce/all_reduce.cpp + ${CMAKE_CURRENT_SOURCE_DIR}/all_reduce/all_reduce_pybind.cpp + ${CMAKE_CURRENT_SOURCE_DIR}/all_reduce/device/all_reduce_op.cpp + CACHE INTERNAL + "CCL Experimental sources to reuse in ttnn build" +)