typo

mlcommons · Sep 28, 2023 · b2f8ff9 · b2f8ff9
1 parent 6cf192a
commit b2f8ff9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/submission_runner.py b/submission_runner.py
@@ -49,7 +49,7 @@
 tf.config.set_visible_devices([], 'GPU')
 
 # disable only for deepspeech if it works fine for other workloads.
-os.environ['XLA_FLAGS'] = '--xla_gpu_enable_triton_gemm=falseos.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:256"# TODO(znado): make a nicer registry of workloads that lookup in.
+os.environ['XLA_FLAGS'] = '--xla_gpu_enable_triton_gemm=false'
 os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:256'
 
 # Workload_path will be appended by '_pytorch' or '_jax' automatically.