fix(cmake): fix ucx cuda path

GitOrigin-RevId: 529f121aa405c71402c314cf2ee6acdfd84bd132

fix(cmake): fix ucx cuda path
GitOrigin-RevId: 529f121aa405c71402c314cf2ee6acdfd84bd132
fad143f0 · Megvii Engine Team · Xinran Xu · 1d3723e1 · fad143f0
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

cmake/ucx.cmake cmake/ucx.cmake +1 -1

未找到文件。
--- 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}