fix flags (#612)

zrr1999 · web-flow · commit 80d4ce5978a9 · 2026-02-05T17:29:51.000+08:00
diff --git a/engine.py b/engine.py
@@ -2,7 +2,7 @@
 
 import os
 
-os.environ["FLAGS_USE_SYSTEM_ALLOCATOR"] = "1"
+os.environ["FLAGS_use_system_allocator"] = "1"
 os.environ["NVIDIA_TF32_OVERRIDE"] = "0"
 
 import argparse
diff --git a/engineV2.py b/engineV2.py
@@ -39,7 +39,7 @@
 
 from tester.api_config.log_writer import *
 
-os.environ["FLAGS_USE_SYSTEM_ALLOCATOR"] = "1"
+os.environ["FLAGS_use_system_allocator"] = "1"
 os.environ["NVIDIA_TF32_OVERRIDE"] = "0"
 
 VALID_TEST_ARGS = {
