diff --git a/conda/tvm-libs/build.sh b/conda/tvm-libs/build.sh index 3a88fdf652ee..772838e63ac6 100644 --- a/conda/tvm-libs/build.sh +++ b/conda/tvm-libs/build.sh @@ -13,7 +13,7 @@ fi if [ -z "$cuda" ] || [ "$cuda" == "False" ]; then CUDA_OPT="" else - CUDA_OPT="-DUSE_CUDA=ON" + CUDA_OPT="-DUSE_CUDA=ON -DUSE_CUBLAS=ON" fi if [ "$target_platform" == "osx-64" ]; then