From bef2fe2b1dd0b8815c09f545e2e35e2243d16868 Mon Sep 17 00:00:00 2001
From: Terry Kong <terryk@nvidia.com>
Date: Fri, 20 Sep 2024 10:11:07 -0700
Subject: [PATCH] be consistent about echoing default flags

Signed-off-by: Terry Kong <terryk@nvidia.com>
---
 rosetta/rosetta/projects/t5x/xla_flags/t5.env               | 1 +
 rosetta/rosetta/projects/vit/xla_flags/vit-base-highgbs.env | 1 +
 rosetta/rosetta/projects/vit/xla_flags/vit-base.env         | 1 +
 3 files changed, 3 insertions(+)

diff --git a/rosetta/rosetta/projects/t5x/xla_flags/t5.env b/rosetta/rosetta/projects/t5x/xla_flags/t5.env
index a2f19a93f..bd4ae50d5 100644
--- a/rosetta/rosetta/projects/t5x/xla_flags/t5.env
+++ b/rosetta/rosetta/projects/t5x/xla_flags/t5.env
@@ -1,3 +1,4 @@
 set -x
+echo "$0 uses default XLA_FLAGS='${XLA_FLAGS:-}'"
 export XLA_PYTHON_CLIENT_MEM_FRACTION=0.8
 set +x
diff --git a/rosetta/rosetta/projects/vit/xla_flags/vit-base-highgbs.env b/rosetta/rosetta/projects/vit/xla_flags/vit-base-highgbs.env
index a217c6988..45140ed88 100644
--- a/rosetta/rosetta/projects/vit/xla_flags/vit-base-highgbs.env
+++ b/rosetta/rosetta/projects/vit/xla_flags/vit-base-highgbs.env
@@ -1,3 +1,4 @@
 set -x
+echo "$0 uses default XLA_FLAGS='${XLA_FLAGS:-}'"
 export XLA_PYTHON_CLIENT_MEM_FRACTION=0.75
 set +x
diff --git a/rosetta/rosetta/projects/vit/xla_flags/vit-base.env b/rosetta/rosetta/projects/vit/xla_flags/vit-base.env
index dd24fcde6..882c9e9e8 100644
--- a/rosetta/rosetta/projects/vit/xla_flags/vit-base.env
+++ b/rosetta/rosetta/projects/vit/xla_flags/vit-base.env
@@ -1,3 +1,4 @@
 set -x
+echo "$0 uses default XLA_FLAGS='${XLA_FLAGS:-}'"
 export XLA_PYTHON_CLIENT_MEM_FRACTION=0.9
 set +x