Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
build.sh 351 Bytes
Newer Older
rusty1s's avatar
first cuda kernel try  
rusty1s committed
1
2
3
4
5
6 
#!/bin/sh

TORCH=$(python -c "import os; import torch; print(os.path.dirname(torch.__file__))")
SRC_DIR=torch_scatter/kernel
BUILD_DIR=torch_scatter/build
rusty1s's avatar
kernel call needs ifdef cplusplus and header include in cu  
rusty1s committed
7 
rm -rf $BUILD_DIR
rusty1s's avatar
first cuda kernel try  
rusty1s committed
8
9 
mkdir -p $BUILD_DIR
nvcc -c -o $BUILD_DIR/kernel.so $SRC_DIR/kernel.cu -Xcompiler -fPIC -shared -I$TORCH/lib/include/TH -I$TORCH/lib/include/THC -I$TORCH/lib/include -I$SRC_DIR