saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[wasm] fix memory info creation (#7461)

Browse source
This commit is contained in:
Yulong Wang 2021-04-27 09:29:21 -07:00 committed by GitHub
parent 4cbd2cce9b
commit c2418a1f42
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/wasm/api.cc
View file

@ -63,7 +63,7 @@ OrtValue* OrtCreateTensor(int data_type, void* data, size_t data_length, size_t*
shapes[i] = dims[i];
}
return Ort::Value::CreateTensor({},
return Ort::Value::CreateTensor(Ort::MemoryInfo::CreateCpu(OrtDeviceAllocator, OrtMemTypeDefault),
data,
data_length,
dims_length > 0 ? shapes.data() : nullptr,