Add CMakeLists.txt binary_dir (#7019)
This commit is contained in:
@@ -330,7 +330,10 @@ endif()
|
|||||||
set(MSCCLPP_USE_CUDA ON)
|
set(MSCCLPP_USE_CUDA ON)
|
||||||
set(MSCCLPP_BYPASS_GPU_CHECK ON)
|
set(MSCCLPP_BYPASS_GPU_CHECK ON)
|
||||||
set(MSCCLPP_BUILD_TESTS OFF)
|
set(MSCCLPP_BUILD_TESTS OFF)
|
||||||
add_subdirectory(${repo-mscclpp_SOURCE_DIR})
|
add_subdirectory(
|
||||||
|
${repo-mscclpp_SOURCE_DIR}
|
||||||
|
${CMAKE_CURRENT_BINARY_DIR}/mscclpp-build
|
||||||
|
)
|
||||||
target_link_libraries(common_ops PRIVATE ${TORCH_LIBRARIES} c10 cuda cublas cublasLt mscclpp_static)
|
target_link_libraries(common_ops PRIVATE ${TORCH_LIBRARIES} c10 cuda cublas cublasLt mscclpp_static)
|
||||||
|
|
||||||
# flash attention
|
# flash attention
|
||||||
|
|||||||
Reference in New Issue
Block a user