clean it up like ryan wants it

Author: aciddelgado

src/models/debugging.cpp

@@ -88,11 +88,23 @@ void DumpTensor(const Model& model, std::ostream& stream, OrtValue* value, bool
     if (dump_value) {
       DumpValues(stream, type_info->GetElementType(), value->GetTensorRawData(), element_count);
     }
-  } else if (memory_info.GetDeviceType() == OrtMemoryInfoDeviceType_GPU || memory_info.GetDeviceType() == 4) {
-    if (memory_info.GetDeviceType() == OrtMemoryInfoDeviceType_GPU)
-      stream << "GPU\r\n";
-    else
-      stream << "DML\r\n";
+    // Internally there are 5 device types defined in onnxruntime but only 3 are exposed in the public API
+    // https://github.com/microsoft/onnxruntime/blob/9dbfee91ca9c2ba2074d19805bb6dedccedbcfe3/include/onnxruntime/core/framework/ortdevice.h#L15
+  } else if (memory_info.GetDeviceType() < 5) {
+    switch (model.p_device_->GetType()) {
+      case DeviceType::CUDA:
+        stream << "CUDA\r\n";
+        break;
+      case DeviceType::DML:
+        stream << "DML\r\n";
+        break;
+      case DeviceType::QNN:
+        stream << "QNN\r\n";
+        break;
+      default:
+        stream << "Unknown\r\n";
+        break;
+    }
     auto type = type_info->GetElementType();
     auto tensor_span = std::span<uint8_t>{const_cast<OrtValue*>(value)->GetTensorMutableData<uint8_t>(), SizeOf(type) * element_count};
     auto device_span = model.p_device_->WrapMemory<uint8_t>(tensor_span);