Fix CI (#9012)

2025-08-09 13:33:42 -07:00
parent 41d71ca488
commit 9a44b643c6
9 changed files with 24 additions and 20 deletions
--- a/test/srt/test_utils_update_weights.py
+++ b/test/srt/test_utils_update_weights.py
@@ -83,7 +83,6 @@ class TestUtilsUpdateWeights(unittest.TestCase):
        # Set up environment variables
        os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
        os.environ["NCCL_CUMEM_ENABLE"] = "0"
-        os.environ["TORCH_NCCL_AVOID_RECORD_STREAMS"] = "1"
        os.environ["CUDA_DEVICE_MAX_CONNECTIONS"] = "4"
        os.environ["CUDA_MODULE_LOADING"] = "AUTO"