Skip to content

Commit

Permalink
[xla:gpu] Disable Conditions in cuda graphs by default
Browse files Browse the repository at this point in the history
Reverts 7466e74

PiperOrigin-RevId: 681058491
  • Loading branch information
IllogicalMoose authored and Google-ML-Automation committed Oct 1, 2024
1 parent e1b38f8 commit 43a6d82
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 6 deletions.
1 change: 0 additions & 1 deletion xla/debug_options_flags.cc
Original file line number Diff line number Diff line change
Expand Up @@ -116,7 +116,6 @@ DebugOptions DefaultDebugOptionsIgnoringFlags() {
opts.add_xla_gpu_enable_command_buffer(DebugOptions::CUBLAS);
opts.add_xla_gpu_enable_command_buffer(DebugOptions::CUSTOM_CALL);
opts.add_xla_gpu_enable_command_buffer(DebugOptions::CUDNN);
opts.add_xla_gpu_enable_command_buffer(DebugOptions::CONDITIONALS);
opts.set_xla_gpu_graph_min_graph_size(5);
opts.set_xla_gpu_graph_enable_concurrent_region(false);
opts.set_xla_cmd_buffer_trace_cache_size(16);
Expand Down
5 changes: 0 additions & 5 deletions xla/service/gpu/transforms/command_buffer_scheduling.cc
Original file line number Diff line number Diff line change
Expand Up @@ -777,11 +777,6 @@ absl::StatusOr<bool> CommandBufferScheduling::Run(
device_description_.driver_version()) <
se::SemanticVersion{12, 3, 0}) {
erase(kRequireTracing); // cuStreamBeginCaptureToGraph
}
if (std::min(device_description_.runtime_version(),
device_description_.driver_version()) <
se::SemanticVersion{12, 4, 0}) {
// Conditionals With Memsets require cuda 12.4.1.
erase(kRequireConditionals); // on-device control flow
}
};
Expand Down

0 comments on commit 43a6d82

Please sign in to comment.