Commit 2d253c11 authored by rusty1s's avatar rusty1s
Browse files

better compiler options

parent b94ecee8
......@@ -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
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