Updated DL device name for MAIA

microsoft · Feb 22, 2024 · 58c51e4 · 58c51e4
1 parent 130fed8
commit 58c51e4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/onnxruntime/core/dlpack/dlpack_converter.cc b/onnxruntime/core/dlpack/dlpack_converter.cc
@@ -86,7 +86,7 @@ OrtDevice GetOrtDevice(const DLDevice& device) {
     case DLDeviceType::kDLCUDA:
     case DLDeviceType::kDLROCM:
       return OrtDevice(OrtDevice::GPU, OrtDevice::MemType::DEFAULT, static_cast<OrtDevice::DeviceId>(device.device_id));
-    case DLDeviceType::kDLExtDev:
+    case DLDeviceType::kDLMAIA:
       return OrtDevice(OrtDevice::NPU, OrtDevice::MemType::DEFAULT, static_cast<OrtDevice::DeviceId>(device.device_id));
     default:
       ORT_THROW("Unsupported device type");
@@ -202,7 +202,7 @@ DLDevice GetDlpackDevice(const OrtValue& ort_value, const int64_t& device_id) {
       break;
     case OrtDevice::FPGA:
     case OrtDevice::NPU:
-      device.device_type = DLDeviceType::kDLExtDev;
+      device.device_type = DLDeviceType::kDLMAIA;
     default:
       ORT_THROW("Cannot pack tensors on this device.");
   }