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