Skip to content

Commit

Permalink
[fix-up] nightly.sh: define BM_DIR earlier in the file (pytorch#6487)
Browse files Browse the repository at this point in the history
(This is a fix-up for pytorch#6485.)

Move the definition up so that we can find ${BM_DIR}/llama.py.
  • Loading branch information
cota authored and amithrm committed Mar 1, 2024
1 parent 0347007 commit 3b0f818
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion benchmarks/nightly.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down

0 comments on commit 3b0f818

Please sign in to comment.