diff options
author | Sadaf Ebrahimi <sadafebrahimi@google.com> | 2022-10-14 20:46:28 +0000 |
---|---|---|
committer | Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com> | 2022-10-14 20:46:28 +0000 |
commit | 0d07658cd9f51af7d98b6d9512deb768a686db3e (patch) | |
tree | 5d2b98a63d4f942bd3babd266cb81763bb59e1dc /analysis/tensorflow/fast_em.sh | |
parent | 912f8cc3fe4b53ebe8931f8f593e18a4f07c96b6 (diff) | |
parent | 6ceb25daa2821e98f7c663bd4bce9ade878b5c73 (diff) | |
download | rappor-0d07658cd9f51af7d98b6d9512deb768a686db3e.tar.gz |
Writing METADATA file for rappor am: 9d7cfa8217 am: 6ceb25daa2android-u-beta-1-gpl
Original change: https://android-review.googlesource.com/c/platform/external/rappor/+/2221504
Change-Id: I3a60682b881d6347f74d55d1d70636a080440cd0
Signed-off-by: Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>
Diffstat (limited to 'analysis/tensorflow/fast_em.sh')
-rwxr-xr-x | analysis/tensorflow/fast_em.sh | 22 |
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 "$@" |