Skip to content

Commit 210d671

Browse files
committed
remove temporary flag for hlo dumps
1 parent 6b55adf commit 210d671

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

submission_runner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@
5353
# Environment variables
5454
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' # Disables tensorRT, cuda warnings.
5555
# 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'
56+
os.environ['XLA_FLAGS'] = '--xla_gpu_enable_triton_gemm=false'
5757

5858
# TODO(znado): make a nicer registry of workloads that lookup in.
5959
BASE_WORKLOADS_DIR = workloads.BASE_WORKLOADS_DIR

0 commit comments

Comments
 (0)