Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
build.sh 361 Bytes
Newer Older
rusty1s's avatar
initial commit  
rusty1s committed
1
2 
#!/bin/sh
rusty1s's avatar
added kernel boilerplate  
rusty1s committed
3
4 
echo "Compiling kernel..."
rusty1s's avatar
initial commit  
rusty1s committed
5 
TORCH=$(python -c "import os; import torch; print(os.path.dirname(torch.__file__))")
rusty1s's avatar
added kernel boilerplate  
rusty1s committed
6
7 
SRC_DIR=torch_cluster/kernel
BUILD_DIR=torch_cluster/build
rusty1s's avatar
initial commit  
rusty1s committed
8
9 

mkdir -p $BUILD_DIR
rusty1s's avatar
added sparse and dense voxel cluster  
rusty1s committed
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