set toolkit_include
@@ -12,6 +12,12 @@ fi
mkdir build
cd build
-cmake ${CMAKE_ARGS} .. -DUSE_FORTRAN=OFF -DGPU_TARGET="All" -DCMAKE_INSTALL_PREFIX=$PREFIX -DCUDA_ARCH_LIST="$CUDA_ARCH_LIST"
+cmake ${CMAKE_ARGS} .. \
+ -DUSE_FORTRAN=OFF \
+ -DGPU_TARGET="All" \
+ -DCMAKE_INSTALL_PREFIX=$PREFIX \
+ -DCUDA_ARCH_LIST="$CUDA_ARCH_LIST" \
+ -DCUDA_TOOLKIT_INCLUDE=$CUDA_HOME/include
+
make -j${CPU_COUNT}
make install
