added cuda boilerplate

build.py

@@ -16,7 +16,7 @@ extra_objects = []
 with_cuda = False
 
 if torch.cuda.is_available():
-    subprocess.call('./build.sh {}'.format(osp.dirname(torch.__file__)))
+    subprocess.call(['./build.sh', osp.dirname(torch.__file__)])
 
     headers += ['torch_spline_conv/src/cuda.h']
     sources += ['torch_spline_conv/src/cuda.c']

build.sh

@@ -2,6 +2,7 @@
 
 echo "Compiling kernel..."
 
+if [ -z "$1" ]; then TORCH=$(python -c "import os; import torch; print(os.path.dirname(torch.__file__))"); else TORCH="$1"; fi
 SRC_DIR=torch_spline_conv/kernel
 BUILD_DIR=torch_spline_conv/build