diff --git a/onnxruntime/core/framework/debug_node_inputs_outputs_utils.cc b/onnxruntime/core/framework/debug_node_inputs_outputs_utils.cc index 7665a90448..607969cd4c 100644 --- a/onnxruntime/core/framework/debug_node_inputs_outputs_utils.cc +++ b/onnxruntime/core/framework/debug_node_inputs_outputs_utils.cc @@ -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) {