Skip to content

Commit

Permalink
[XLA:GPU] Enable auto while loop double buffering.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 678606953
  • Loading branch information
golechwierowicz authored and Google-ML-Automation committed Sep 25, 2024
1 parent b5216ad commit 41c58a5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion xla/debug_options_flags.cc
Original file line number Diff line number Diff line change
Expand Up @@ -220,7 +220,7 @@ DebugOptions DefaultDebugOptionsIgnoringFlags() {
opts.set_xla_gpu_cudnn_gemm_fusion_level(0);
opts.set_xla_gpu_enable_while_loop_double_buffering(false);
opts.set_xla_gpu_enable_while_loop_unrolling(
DebugOptions::WHILE_LOOP_UNROLLING_NO_UNROLL);
DebugOptions::WHILE_LOOP_UNROLLING_AUTO_UNROLL);
opts.set_xla_gpu_ensure_minor_dot_contraction_dims(false);
opts.set_xla_gpu_filter_kernels_spilling_registers_on_autotuning(true);
opts.set_xla_gpu_llvm_verification_level(0);
Expand Down

0 comments on commit 41c58a5

Please sign in to comment.