[Issue 62] flexible whl for different cuda version

install.sh

@@ -56,7 +56,7 @@ cd 3rdparty/tvm
 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 ../../..
 

setup.py

@@ -176,7 +176,7 @@ def build_tvm(llvm_config_path):
     # Set LLVM path and enable CUDA in config.cmake
     with open("config.cmake", "a") as config_file:
         config_file.write(f"set(USE_LLVM {llvm_config_path})\n")
-        config_file.write("set(USE_CUDA ON)\n")
+        config_file.write("set(USE_CUDA /usr/local/cuda)\n")
     # Run CMake and make
     try:
         subprocess.check_call(["cmake", ".."])