diff --git a/llvm/utils/lit/lit/formats/googletest.py b/llvm/utils/lit/lit/formats/googletest.py --- a/llvm/utils/lit/lit/formats/googletest.py +++ b/llvm/utils/lit/lit/formats/googletest.py @@ -112,8 +112,8 @@ shard_env = { 'GTEST_OUTPUT': 'json:' + test.gtest_json_file, 'GTEST_SHUFFLE': '1' if use_shuffle else '0', - 'GTEST_TOTAL_SHARDS': total_shards, - 'GTEST_SHARD_INDEX': shard_idx + 'GTEST_TOTAL_SHARDS': os.getenv("GTEST_TOTAL_SHARDS", total_shards), + 'GTEST_SHARD_INDEX': os.getenv("GTEST_SHARD_INDEX", shard_idx) } test.config.environment.update(shard_env)