aboutsummaryrefslogtreecommitdiff
path: root/analysis/tensorflow/fast_em.sh
diff options
context:
space:
mode:
authorSadaf Ebrahimi <sadafebrahimi@google.com>2022-10-14 21:19:34 +0000
committerAutomerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>2022-10-14 21:19:34 +0000
commit0a1cff34c489372d3f515bf81efef2f9f5c02561 (patch)
tree5d2b98a63d4f942bd3babd266cb81763bb59e1dc /analysis/tensorflow/fast_em.sh
parent912f8cc3fe4b53ebe8931f8f593e18a4f07c96b6 (diff)
parent0d07658cd9f51af7d98b6d9512deb768a686db3e (diff)
downloadrappor-0a1cff34c489372d3f515bf81efef2f9f5c02561.tar.gz
Original change: https://android-review.googlesource.com/c/platform/external/rappor/+/2221504 Change-Id: I3bc1ac5d5c22eaea2ae44ebf876e3664d89040c0 Signed-off-by: Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>
Diffstat (limited to 'analysis/tensorflow/fast_em.sh')
-rwxr-xr-xanalysis/tensorflow/fast_em.sh22
1 files changed, 22 insertions, 0 deletions
diff --git a/analysis/tensorflow/fast_em.sh b/analysis/tensorflow/fast_em.sh
new file mode 100755
index 0000000..d1ddb79
--- /dev/null
+++ b/analysis/tensorflow/fast_em.sh
@@ -0,0 +1,22 @@
+#!/bin/bash
+#
+# Wrapper to run fast_em.py using TensorFlow configured for a GPU. CUDA
+# environment variables must be set.
+#
+# Usage:
+# ./fast_em.sh <args>
+
+set -o nounset
+set -o pipefail
+set -o errexit
+
+readonly THIS_DIR=$(dirname $0)
+
+fast-em() {
+ # Never returns
+ LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64 \
+ CUDA_HOME=/usr/local/cuda-7.0 \
+ exec $THIS_DIR/fast_em.py "$@"
+}
+
+fast-em "$@"