We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3e02ced commit b663556Copy full SHA for b663556
build.sh
@@ -7,4 +7,4 @@ SRC_DIR=torch_scatter/kernel
7
BUILD_DIR=torch_scatter/build
8
9
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"
+$(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