saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[DML EP] Add intermediate tensor dumping for DML (#22246)

Browse source 
### Description
<!-- Describe your changes. -->



### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->
This commit is contained in:
Patrice Vignola 2024-09-27 12:39:45 -07:00 committed by GitHub
parent 6e3163faa5
commit 14ba2fb83c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/core/framework/debug_node_inputs_outputs_utils.cc
View file

@ -333,7 +333,7 @@ void DumpTensor(
} else {
std::cout << tensor_location << "\n";
#if defined(USE_CUDA) || defined(USE_ROCM)
#if defined(USE_CUDA) || defined(USE_ROCM) || defined(USE_DML)
const auto data_type = tensor.DataType();
// Dumping GPU only when cuda is enabled.
if (tensor_location.device.Type() == OrtDevice::GPU) {