1*da0073e9SAndroid Build Coastguard Workerif(NOT __NCCL_INCLUDED) 2*da0073e9SAndroid Build Coastguard Worker set(__NCCL_INCLUDED TRUE) 3*da0073e9SAndroid Build Coastguard Worker 4*da0073e9SAndroid Build Coastguard Worker if(USE_SYSTEM_NCCL) 5*da0073e9SAndroid Build Coastguard Worker # NCCL_ROOT, NCCL_LIB_DIR, NCCL_INCLUDE_DIR will be accounted in the following line. 6*da0073e9SAndroid Build Coastguard Worker find_package(rccl REQUIRED) 7*da0073e9SAndroid Build Coastguard Worker if(rccl_FOUND) 8*da0073e9SAndroid Build Coastguard Worker message(STATUS "RCCL Found!") 9*da0073e9SAndroid Build Coastguard Worker add_library(__caffe2_nccl INTERFACE) 10*da0073e9SAndroid Build Coastguard Worker target_link_libraries(__caffe2_nccl INTERFACE ${PYTORCH_RCCL_LIBRARIES}) 11*da0073e9SAndroid Build Coastguard Worker target_include_directories(__caffe2_nccl INTERFACE ${RCCL_INCLUDE_DIRS}) 12*da0073e9SAndroid Build Coastguard Worker else() 13*da0073e9SAndroid Build Coastguard Worker message(STATUS "RCCL NOT Found!") 14*da0073e9SAndroid Build Coastguard Worker endif() 15*da0073e9SAndroid Build Coastguard Worker else() 16*da0073e9SAndroid Build Coastguard Worker message(STATUS "USE_SYSTEM_NCCL=OFF is not supported yet when using RCCL") 17*da0073e9SAndroid Build Coastguard Worker endif() 18*da0073e9SAndroid Build Coastguard Workerendif() 19