update script

flexflow · Feb 11, 2019 · 410effd · 410effd
1 parent a2bc2c8
commit 410effd
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/ffcompile.sh b/ffcompile.sh
@@ -1,5 +1,13 @@
 #!/bin/bash
 
+# GPU_ARCH must be one of fermi, k20, pascal, volta
+export GPU_ARCH=volta
+
+# set CUDA_HOME to the directoy of cuda, which should have subdirectories bin/, include/, and lib/
+export CUDA_HOME=/usr/local/cuda
+git submodule init
+git submodule update
+
 APP="$1"
 
 if [ -z "$APP" ]; then echo "Usage: ./ffcompile app"; exit; fi