| if(NOT __NCCL_INCLUDED) |
| set(__NCCL_INCLUDED TRUE) |
| |
| if(USE_SYSTEM_NCCL) |
| # NCCL_ROOT, NCCL_LIB_DIR, NCCL_INCLUDE_DIR will be accounted in the following line. |
| find_package(rccl REQUIRED) |
| if(rccl_FOUND) |
| message(STATUS "RCCL Found!") |
| add_library(__caffe2_nccl INTERFACE) |
| target_link_libraries(__caffe2_nccl INTERFACE ${PYTORCH_RCCL_LIBRARIES}) |
| target_include_directories(__caffe2_nccl INTERFACE ${RCCL_INCLUDE_DIRS}) |
| else() |
| message(STATUS "RCCL NOT Found!") |
| endif() |
| else() |
| message(STATUS "USE_SYSTEM_NCCL=OFF is not supported yet when using RCCL") |
| endif() |
| endif() |
