Skip to content

Commit bdb3d46

Browse files
shawnwang18tensorflower-gardener
authored andcommitted
PR tensorflow#23426: [XLA:GPU] set CUBLASLT command buffer as default enabled
Imported from GitHub PR openxla/xla#23426 Copybara import of the project: -- 3d2f748916dc6a9c72c681f9bd9fc46f03d396ee by Shawn Wang <shawnw@nvidia.com>: set CUBLASLT command buffer as default enabled Merging this change closes tensorflow#23426 PiperOrigin-RevId: 734372857
1 parent 3302bdf commit bdb3d46

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

third_party/xla/xla/debug_options_flags.cc

+1
Original file line numberDiff line numberDiff line change
@@ -116,6 +116,7 @@ DebugOptions DefaultDebugOptionsIgnoringFlags() {
116116

117117
opts.add_xla_gpu_enable_command_buffer(DebugOptions::FUSION);
118118
opts.add_xla_gpu_enable_command_buffer(DebugOptions::CUBLAS);
119+
opts.add_xla_gpu_enable_command_buffer(DebugOptions::CUBLASLT);
119120
opts.add_xla_gpu_enable_command_buffer(DebugOptions::CUSTOM_CALL);
120121
opts.add_xla_gpu_enable_command_buffer(DebugOptions::CUDNN);
121122
opts.set_xla_gpu_graph_min_graph_size(5);

0 commit comments

Comments
 (0)