#!/bin/sh

echo "Compiling kernel..."

SRC_DIR=torch_spline_conv/kernel
BUILD_DIR=torch_spline_conv/build

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"