From fd7f0c3cfcafebc56e0b11e100a1ec2cb4a723b5 Mon Sep 17 00:00:00 2001 From: Patrice Vignola Date: Wed, 12 Apr 2023 16:44:53 -0700 Subject: [PATCH] [DML EP] Use ORT node names in DML execution plans (#15411) --- .../dml/DmlExecutionProvider/src/Operators/DmlOperator.cpp | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlOperator.cpp b/onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlOperator.cpp index bbd57b4d98..25c7be42d6 100644 --- a/onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlOperator.cpp +++ b/onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlOperator.cpp @@ -194,6 +194,11 @@ namespace Dml DML_EXECUTION_FLAGS executionFlags = GetExecutionFlags(); ORT_THROW_IF_FAILED(dmlDevice1->CompileGraph(&graphDesc, executionFlags, IID_PPV_ARGS(&m_compiledOperator))); + // Static buffer (might truncate name) to avoid excessive dynamic allocation only for debugging purposes. + wchar_t nodeName[512]; + ORT_THROW_IF_FAILED(kernelInfo.GetNodeWrapperInterface()->GetWideName(sizeof(nodeName), nodeName)); + ORT_THROW_IF_FAILED(m_compiledOperator->SetName(nodeName)); + UINT64 persistentResourceSize = m_compiledOperator->GetBindingProperties().PersistentResourceSize; if (persistentResourceSize > 0) {