Build cuda packages with cublas support

apache · Mar 7, 2019 · 80ba94a · 80ba94a
1 parent d630e4c
commit 80ba94a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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