Skip to content

Commit 2d253c1

Browse files
committed
better compiler options
1 parent b94ecee commit 2d253c1

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=aten/THC
77
BUILD_DIR=torch_cluster/_ext
88

99
mkdir -p $BUILD_DIR
10-
$(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"
10+
$(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

0 commit comments

Comments
 (0)