better compiler options

rusty1s · Apr 30, 2018 · 2d253c1 · 2d253c1
1 parent b94ecee
commit 2d253c1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/build.sh b/build.sh
@@ -7,4 +7,4 @@ SRC_DIR=aten/THC
 BUILD_DIR=torch_cluster/_ext
 
 mkdir -p $BUILD_DIR
-$(which nvcc) -c -o "$BUILD_DIR/THC.so" "$SRC_DIR/THC.cu" -arch=sm_52 -Xcompiler -fPIC -shared "-I$TORCH/lib/include/TH" "-I$TORCH/lib/include" "-I$SRC_DIR"
+$(which nvcc) "-I$TORCH/lib/include" "-I$TORCH/lib/include/TH" "-I$TORCH/lib/include/THC" "-I$SRC_DIR" -c "$SRC_DIR/THC.cu" -o "$BUILD_DIR/THC.o" --compiler-options '-fPIC' -std=c++11