removed triton gemm related flag

NVIDIA · Aug 21, 2024 · e74a333 · e74a333
1 parent 4515ead
commit e74a333
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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