From fad143f0a88495113aac897d7a7f2ec09ae9a18c Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Wed, 15 Apr 2020 13:39:04 +0800
Subject: [PATCH] fix(cmake): fix ucx cuda path

GitOrigin-RevId: 529f121aa405c71402c314cf2ee6acdfd84bd132
---
 cmake/ucx.cmake | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cmake/ucx.cmake b/cmake/ucx.cmake
index 0f18752..0f46c4f 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}
-- 
GitLab