Commit b663556a authored by rusty1s's avatar rusty1s
Browse files

better compiler options

parent 3e02ced3
...@@ -7,4 +7,4 @@ SRC_DIR=torch_scatter/kernel ...@@ -7,4 +7,4 @@ SRC_DIR=torch_scatter/kernel
BUILD_DIR=torch_scatter/build BUILD_DIR=torch_scatter/build
mkdir -p $BUILD_DIR mkdir -p $BUILD_DIR
$(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
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment