saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-01 23:30:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Ortmemoryinfo struct changed

Browse source
This commit is contained in:
Ryan Lai 2020-01-16 11:05:49 -08:00
parent 908372b468
commit 650bdf9533
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
winml/adapter/WinMLAdapter.cpp
View file

@ -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<onnxruntime::Tensor>();
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;
}