提交 fad143f0 编写于 作者: M Megvii Engine Team 提交者: Xinran Xu

fix(cmake): fix ucx cuda path

GitOrigin-RevId: 529f121aa405c71402c314cf2ee6acdfd84bd132
上级 1d3723e1
...@@ -66,7 +66,7 @@ ExternalProject_add( ...@@ -66,7 +66,7 @@ ExternalProject_add(
SOURCE_DIR ${UCX_DIR} SOURCE_DIR ${UCX_DIR}
PREFIX ${UCX_BUILD_DIR} PREFIX ${UCX_BUILD_DIR}
CONFIGURE_COMMAND ${UCX_DIR}/configure --enable-mt --with-pic --disable-static 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 BUILD_COMMAND ${MAKE_COMMAND} all
INSTALL_COMMAND ${MAKE_COMMAND} install INSTALL_COMMAND ${MAKE_COMMAND} install
BUILD_BYPRODUCTS ${UCP_LIB} ${UCT_LIB} ${UCS_LIB} ${UCM_LIB} BUILD_BYPRODUCTS ${UCP_LIB} ${UCT_LIB} ${UCS_LIB} ${UCM_LIB}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册