diff --git a/cmake/ucx.cmake b/cmake/ucx.cmake index 0f187529b694d7514e55d3d9f78ffa6af99c7a8f..0f46c4f6803802f36c87667ba9e97c53754d5619 100644 --- a/cmake/ucx.cmake +++ b/cmake/ucx.cmake @@ -66,7 +66,7 @@ ExternalProject_add( SOURCE_DIR ${UCX_DIR} PREFIX ${UCX_BUILD_DIR} CONFIGURE_COMMAND ${UCX_DIR}/configure --enable-mt --with-pic --disable-static - --with-cuda --disable-numa --prefix=${UCX_BUILD_DIR} ${UCX_CONFIGURE_ARGS} + --with-cuda=${CUDA_HOME} --disable-numa --prefix=${UCX_BUILD_DIR} ${UCX_CONFIGURE_ARGS} BUILD_COMMAND ${MAKE_COMMAND} all INSTALL_COMMAND ${MAKE_COMMAND} install BUILD_BYPRODUCTS ${UCP_LIB} ${UCT_LIB} ${UCS_LIB} ${UCM_LIB}