specify gpu architecture level

d10668f9 · rusty1s · 98327f62 · d10668f9
Commit d10668f9 authored Jan 10, 2018 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

build.sh build.sh +1 -1

No files found.
--- a/build.sh
+++ b/build.sh
@@ -5,4 +5,4 @@ SRC_DIR=torch_scatter/kernel
 BUILD_DIR=torch_scatter/build

 mkdir -p $BUILD_DIR
-nvcc -c -o $BUILD_DIR/kernel.so $SRC_DIR/kernel.cu -Xcompiler -fPIC -shared -I$TORCH/lib/include/TH -I$TORCH/lib/include/THC -I$SRC_DIR
+nvcc -c -o $BUILD_DIR/kernel.so $SRC_DIR/kernel.cu -arch=sm_30 -Xcompiler -fPIC -shared -I$TORCH/lib/include/TH -I$TORCH/lib/include/THC -I$SRC_DIR