OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-06 12:20:11 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Improve CUDNN error messages

Browse Source 
PiperOrigin-RevId: 826124080
This commit is contained in:
William S. Moses 2025-10-30 11:46:23 -07:00 committed by TensorFlower Gardener
parent 9eeebc9be5
commit a94890b1f9
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
third_party/xla/xla/stream_executor/cuda/cuda_dnn.cc vendored
View File

@ -5421,6 +5421,11 @@ absl::Status CreateOpRunners(
.setEngineConfig(filtered_configs[i], op_graph->getTag())
.build();
if (plan.get_status() != CUDNN_STATUS_SUCCESS) {
std::string message(65535, '\0');
cudnnGetLastErrorString(message.data(), message.size());
VLOG(4) << "Failed building ExecutionPlan: found error: "
<< cudnnGetErrorString(plan.get_status())
<< " with message: " << message;
continue;
}