fix flags

Author: zrr1999

engine.py

@@ -2,7 +2,8 @@
 
 import os
 
-os.environ["FLAGS_USE_SYSTEM_ALLOCATOR"] = "1"
+os.environ["FLAGS_use_system_allocator"] = "1"
+os.environ["FLAGS_check_cuda_error"] = "1"
 os.environ["NVIDIA_TF32_OVERRIDE"] = "0"
 
 import argparse

engineV2.py

@@ -39,7 +39,8 @@
 
 from tester.api_config.log_writer import *
 
-os.environ["FLAGS_USE_SYSTEM_ALLOCATOR"] = "1"
+os.environ["FLAGS_use_system_allocator"] = "1"
+os.environ["FLAGS_check_cuda_error"] = "1"
 os.environ["NVIDIA_TF32_OVERRIDE"] = "0"
 
 VALID_TEST_ARGS = {