Skip to content

Commit

Permalink
[xla:gpu] Make command buffer warnings less noisy
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 595564116
  • Loading branch information
ezhulenev authored and tensorflower-gardener committed Jan 4, 2024
1 parent 1e406f7 commit 3491e63
Showing 1 changed file with 11 additions and 10 deletions.
21 changes: 11 additions & 10 deletions third_party/xla/xla/service/gpu/command_buffer_scheduling.cc
Original file line number Diff line number Diff line change
Expand Up @@ -537,18 +537,19 @@ StatusOr<bool> CommandBufferScheduling::Run(
auto erase = [&](absl::Span<const DebugOptions::CommandBufferCmdType> cmds) {
for (auto cmd : cmds) {
if (config.erase(cmd)) {
LOG(WARNING) << "Removed command buffer support for "
<< DebugOptions::CommandBufferCmdType_Name(cmd)
<< " as it's not supported with gpu toolkit version "
<< gpu_toolkit_version_ << " and driver version "
<< gpu_driver_version_
<< ". This might negatively impact peformance. To enable "
<< DebugOptions::CommandBufferCmdType_Name(cmd)
<< " support in command buffers use cuda-compat package: "
LOG_FIRST_N(WARNING, 10)
<< "Removed command buffer support for "
<< DebugOptions::CommandBufferCmdType_Name(cmd)
<< " as it's not supported with gpu toolkit version "
<< gpu_toolkit_version_ << " and driver version "
<< gpu_driver_version_
<< ". This might negatively impact peformance. To enable "
<< DebugOptions::CommandBufferCmdType_Name(cmd)
<< " support in command buffers use cuda-compat package: "
#if defined(PLATFORM_GOOGLE)
<< "set CUDA_COMPAT_LOAD=1 env variable.";
<< "set CUDA_COMPAT_LOAD=1 env variable.";
#else
<< "https://docs.nvidia.com/deploy/cuda-compatibility/.";
<< "https://docs.nvidia.com/deploy/cuda-compatibility/.";
#endif
}
}
Expand Down

0 comments on commit 3491e63

Please sign in to comment.