From 3b0f818bde8b1105d14a940e38c748f0394aa659 Mon Sep 17 00:00:00 2001
From: Emilio Cota <ecg@google.com>
Date: Tue, 6 Feb 2024 17:26:13 -0500
Subject: [PATCH] [fix-up] nightly.sh: define BM_DIR earlier in the file
 (#6487)

(This is a fix-up for #6485.)

Move the definition up so that we can find ${BM_DIR}/llama.py.
---
 benchmarks/nightly.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/benchmarks/nightly.sh b/benchmarks/nightly.sh
index 515814ef8f52..98c9a64495fe 100755
--- a/benchmarks/nightly.sh
+++ b/benchmarks/nightly.sh
@@ -43,6 +43,7 @@ if [[ ${ENABLE_PROFILING?} ]]; then
   NIGHTLY_RESULTS=nightly_profiling_results
 fi
 WORKSPACE_DIR=${OUTPUT_DIR:?}/${NIGHTLY_RUNS:?}/${WORKSPACE:?}
+BM_DIR=${WORKSPACE_DIR:?}/pytorch/xla/benchmarks
 
 # Intermediate results, which are processed to generate reports.
 WORKSPACE_RESULTS_DIR=${WORKSPACE_DIR:?}/experiment_results
@@ -176,7 +177,6 @@ cp ${WORKSPACE_RESULTS_DIR:?}/results.jsonl \
 PYTORCH_GIT_REV=$(git -C pytorch rev-parse --short HEAD)
 XLA_GIT_TAG=$(git -C pytorch/xla describe --tags --always)
 GIT_TAGS="PT: ${PYTORCH_GIT_REV:?} XLA: ${XLA_GIT_TAG:?}"
-BM_DIR=${WORKSPACE_DIR:?}/pytorch/xla/benchmarks
 
 COMMON_TITLE_PREFIX=
 if [[ ${ENABLE_PROFILING?} ]]; then