mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
cda210e23b
- Modifies the current cmake build definitions to use `find_package` to find UCX and UCC installed in the system - Install UCX and UCC in CUDA dockers - Build PyTorch with `USE_UCC=1` in pipelines - Currently, we are not running unit tests with the UCC PG. Those tests will be added in future PRs. Pull Request resolved: https://github.com/pytorch/pytorch/pull/81583 Approved by: https://github.com/vtlam, https://github.com/malfet
16 lines
485 B
CMake
16 lines
485 B
CMake
if(NOT __UCC_INCLUDED)
|
|
set(__UCC_INCLUDED TRUE)
|
|
|
|
if(USE_SYSTEM_UCC)
|
|
find_package(UCC REQUIRED)
|
|
find_package(UCX REQUIRED)
|
|
if(UCC_FOUND AND UCX_FOUND)
|
|
add_library(__caffe2_ucc INTERFACE)
|
|
target_link_libraries(__caffe2_ucc INTERFACE ucx::ucs ucx::ucp ucc::ucc)
|
|
target_include_directories(__caffe2_ucc INTERFACE ${UCC_INCLUDE_DIRS})
|
|
endif()
|
|
else()
|
|
message(FATAL_ERROR "USE_SYSTEM_UCC=OFF is not supported yet when using UCC")
|
|
endif()
|
|
endif()
|