1*da0073e9SAndroid Build Coastguard Workerif(NOT __UCC_INCLUDED) 2*da0073e9SAndroid Build Coastguard Worker set(__UCC_INCLUDED TRUE) 3*da0073e9SAndroid Build Coastguard Worker 4*da0073e9SAndroid Build Coastguard Worker if(USE_SYSTEM_UCC) 5*da0073e9SAndroid Build Coastguard Worker find_package(UCC REQUIRED) 6*da0073e9SAndroid Build Coastguard Worker find_package(UCX REQUIRED) 7*da0073e9SAndroid Build Coastguard Worker if(UCC_FOUND AND UCX_FOUND) 8*da0073e9SAndroid Build Coastguard Worker add_library(__caffe2_ucc INTERFACE) 9*da0073e9SAndroid Build Coastguard Worker target_link_libraries(__caffe2_ucc INTERFACE ucx::ucs ucx::ucp ucc::ucc) 10*da0073e9SAndroid Build Coastguard Worker target_include_directories(__caffe2_ucc INTERFACE ${UCC_INCLUDE_DIRS}) 11*da0073e9SAndroid Build Coastguard Worker endif() 12*da0073e9SAndroid Build Coastguard Worker else() 13*da0073e9SAndroid Build Coastguard Worker message(FATAL_ERROR "USE_SYSTEM_UCC=OFF is not supported yet when using UCC") 14*da0073e9SAndroid Build Coastguard Worker endif() 15*da0073e9SAndroid Build Coastguard Workerendif() 16