Merge branch 'main' of https://github.com/microsoft/BitBLAS into tl-test

microsoft · Sep 27, 2024 · 8d45157 · 8d45157
2 parents 6a07890 + a912e02
commit 8d45157
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/maint/scripts/installation.sh b/maint/scripts/installation.sh
@@ -46,17 +46,20 @@ fi
 
 echo "Download and extraction completed successfully."
 
-LLVM_CONFIG_PATH="${EXTRACT_PATH}/$(basename ${FILE_NAME} .tar.xz)/bin/llvm-config"
+LLVM_CONFIG_PATH="$(realpath ${EXTRACT_PATH}/$(basename ${FILE_NAME} .tar.xz)/bin/llvm-config)"
 echo "LLVM config path: $LLVM_CONFIG_PATH"
 
 # clone and build tvm
 git submodule update --init --recursive
 
 cd 3rdparty/tvm
+if [ -d build ]; then
+    rm -rf build
+fi
 mkdir build
 cp cmake/config.cmake build
 cd build
-echo "set(USE_LLVM $LLVM_CONFIG_PATH)" >> config.cmake && echo "set(USE_CUDA ON)" >> config.cmake
+echo "set(USE_LLVM $LLVM_CONFIG_PATH)" >> config.cmake && echo "set(USE_CUDA /usr/local/cuda)" >> config.cmake
 
 cmake .. && make -j && cd ../../..