be consistent about echoing default flags

Signed-off-by: Terry Kong <terryk@nvidia.com>
NVIDIA · Sep 20, 2024 · bef2fe2 · bef2fe2
1 parent 550d38d
commit bef2fe2
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git 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
@@ -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
@@ -1,3 +1,4 @@
 set -x
+echo "$0 uses default XLA_FLAGS='${XLA_FLAGS:-}'"
 export XLA_PYTHON_CLIENT_MEM_FRACTION=0.9
 set +x