Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
build.sh 288 Bytes
Newer Older
rusty1s's avatar
clean up  
rusty1s committed
1
2
3
4
5
6
7 
#!/bin/sh

echo "Compiling kernel..."

SRC_DIR=torch_spline_conv/kernel
BUILD_DIR=torch_spline_conv/build
rusty1s's avatar
pass torch path  
rusty1s committed
8
9 
mkdir -p "$BUILD_DIR"
$(which nvcc) -c -o "$BUILD_DIR/kernel.so" "$SRC_DIR/kernel.cu" -arch=sm_35 -Xcompiler -fPIC -shared "-I$1/lib/include/TH" "-I$1/lib/include/THC" "-I$SRC_DIR"