Skip to content

Commit b663556

Browse files
committed
better compiler options
1 parent 3e02ced commit b663556

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

build.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,4 +7,4 @@ SRC_DIR=torch_scatter/kernel
77
BUILD_DIR=torch_scatter/build
88

99
mkdir -p $BUILD_DIR
10-
$(which nvcc) -c -o "$BUILD_DIR/kernel.so" "$SRC_DIR/kernel.cu" -arch=sm_35 -Xcompiler -fPIC -shared "-I$TORCH/lib/include/TH" "-I$TORCH/lib/include/THC" "-I$SRC_DIR"
10+
$(which nvcc) "-I$TORCH/lib/include" "-I$TORCH/lib/include/TH" "-I$TORCH/lib/include/THC" "-I$SRC_DIR" -c "$SRC_DIR/kernel.cu" -o "$BUILD_DIR/kernel.o" --compiler-options '-fPIC' -std=c++11

0 commit comments

Comments
 (0)