From e74a333f63c1ef77db3f446a60fe3d5ddecddac1 Mon Sep 17 00:00:00 2001 From: Md Fahim Faysal Khan Date: Wed, 21 Aug 2024 14:21:20 -0700 Subject: [PATCH] removed triton gemm related flag --- .github/container/test-maxtext.sh | 1 - 1 file changed, 1 deletion(-) diff --git a/.github/container/test-maxtext.sh b/.github/container/test-maxtext.sh index 21591c91c..39f82246b 100755 --- a/.github/container/test-maxtext.sh +++ b/.github/container/test-maxtext.sh @@ -224,7 +224,6 @@ export XLA_PYTHON_CLIENT_MEM_FRACTION=${MEM_FRACTION} export CUDA_DEVICE_MAX_CONNECTIONS=1 export BASE_XLA_FLAGS=${BASE_XLA_FLAGS:---xla_gpu_enable_latency_hiding_scheduler=true - --xla_gpu_enable_triton_gemm=false --xla_gpu_graph_level=0 --xla_gpu_enable_highest_priority_async_stream=true --xla_gpu_all_reduce_combine_threshold_bytes=1073741824