diff --git a/winml/adapter/WinMLAdapter.cpp b/winml/adapter/WinMLAdapter.cpp index 56e7cc043e..eef56c6079 100644 --- a/winml/adapter/WinMLAdapter.cpp +++ b/winml/adapter/WinMLAdapter.cpp @@ -594,7 +594,7 @@ class WinMLAdapter : public Microsoft::WRL::RuntimeClass< auto allocator = provider->GetAllocator(0, ::OrtMemType::OrtMemTypeDefault); const auto& info = allocator->Info(); - *memory_info = new OrtMemoryInfo(info.name, info.type, info.device, info.id, info.mem_type); + *memory_info = new OrtMemoryInfo(info.name, info.alloc_type, info.device, info.id, info.mem_type); if (*memory_info == nullptr) { return E_OUTOFMEMORY; } @@ -605,7 +605,7 @@ class WinMLAdapter : public Microsoft::WRL::RuntimeClass< HRESULT STDMETHODCALLTYPE GetValueMemoryInfo(const OrtValue* ort_value, OrtMemoryInfo** memory_info) override try { const auto& tensor = ort_value->Get(); auto info = tensor.Location(); - *memory_info = new OrtMemoryInfo(info.name, info.type, info.device, info.id, info.mem_type); + *memory_info = new OrtMemoryInfo(info.name, info.alloc_type, info.device, info.id, info.mem_type); if (*memory_info == nullptr) { return E_OUTOFMEMORY; }