xref: /aosp_15_r20/external/pytorch/cmake/External/ucc.cmake (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
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