xref: /aosp_15_r20/external/rappor/analysis/tensorflow/fast_em.sh (revision 2abb31345f6c95944768b5222a9a5ed3fc68cc00)
1#!/bin/bash
2#
3# Wrapper to run fast_em.py using TensorFlow configured for a GPU.  CUDA
4# environment variables must be set.
5#
6# Usage:
7#   ./fast_em.sh <args>
8
9set -o nounset
10set -o pipefail
11set -o errexit
12
13readonly THIS_DIR=$(dirname $0)
14
15fast-em() {
16  # Never returns
17  LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64 \
18  CUDA_HOME=/usr/local/cuda-7.0 \
19    exec $THIS_DIR/fast_em.py "$@"
20}
21
22fast-em "$@"
23