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