We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 6b55adf commit 210d671Copy full SHA for 210d671
submission_runner.py
@@ -53,7 +53,7 @@
53
# Environment variables
54
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' # Disables tensorRT, cuda warnings.
55
# disable only for deepspeech if it works fine for other workloads
56
-os.environ['XLA_FLAGS'] = '--xla_gpu_enable_triton_gemm=false --xla_dump_to=/logs/xla_dump_jax_lm_10_06_bsz64_seq1028 --xla_dump_hlo_as_proto'
+os.environ['XLA_FLAGS'] = '--xla_gpu_enable_triton_gemm=false'
57
58
# TODO(znado): make a nicer registry of workloads that lookup in.
59
BASE_WORKLOADS_DIR = workloads.BASE_WORKLOADS_DIR
0 commit comments