saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-11 17:48:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[JSEP] allow JsCustomAllocator to deal with zero sized input (#17660)

Browse source 
### Description
allow JsCustomAllocator to deal with zero sized input.

This is a standalone fix for zero-sized tensor handling for
JsCustomAllocator. There are other components in JSEP not supporting
zero-sized tensors need to be fixed.
This commit is contained in:
Yulong Wang 2023-09-25 12:20:56 -07:00 committed by GitHub
parent 905faea3b2
commit fcfc2391b8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
onnxruntime/core/providers/js/allocator.cc
View file

@ -10,6 +10,10 @@ namespace onnxruntime {
namespace js {
void* JsCustomAllocator::Alloc(size_t size) {
if (size == 0) {
return nullptr;
}
void* p = EM_ASM_PTR({ return Module.jsepAlloc($0); }, size);
stats_.num_allocs++;
stats_.bytes_in_use += size;
@ -17,8 +21,10 @@ void* JsCustomAllocator::Alloc(size_t size) {
}
void JsCustomAllocator::Free(void* p) {
size_t size = (size_t)(void*)EM_ASM_PTR({ return Module.jsepFree($0); }, p);
stats_.bytes_in_use -= size;
if (p != nullptr) {
size_t size = (size_t)(void*)EM_ASM_PTR({ return Module.jsepFree($0); }, p);
stats_.bytes_in_use -= size;
}
}
void JsCustomAllocator::GetStats(AllocatorStats* stats) {