Fix cuda support detection when passed UCX_DIR

This commit is contained in:
Christoph Niethammer 2022-11-11 08:37:34 +00:00
parent 026e9ba30b
commit 3ff9e1617c

View file

@ -50,7 +50,7 @@ fi
if [ ! -z $CUDA_DIR ] ; then if [ ! -z $CUDA_DIR ] ; then
[ -z "$UCX_DIR" ] && sit_fail "Have to be configured with external UCX" [ -z "$UCX_DIR" ] && sit_fail "Have to be configured with external UCX"
ucx_info -b | grep -e "HAVE_CUDA *1" >/dev/null || sit_fail "UCX must be built with CUDA support" $UCX_DIR/bin/ucx_info -b | grep -e "HAVE_CUDA *1" >/dev/null || sit_fail "UCX must be built with CUDA support"
CONFIGURE_OPTS+=" --with-cuda=$CUDA_DIR" CONFIGURE_OPTS+=" --with-cuda=$CUDA_DIR"
fi fi