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

Avoid to call memory allocation if tensor size is 0 (#972)

Browse source
This commit is contained in:
Yufeng Li 2019-05-05 21:46:00 -07:00 committed by GitHub
parent af4a41fd13
commit 8b3c8561ce
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/core/framework/tensor.cc
View file

@ -21,7 +21,9 @@ Tensor::Tensor(MLDataType p_type, const TensorShape& shape, std::shared_ptr<IAll
int64_t shape_size = shape.Size();
if (shape_size < 0 || static_cast<uint64_t>(shape_size) >= std::numeric_limits<size_t>::max())
ORT_THROW("shape.Size() must >=0");
void* p_data = allocator->AllocArray(static_cast<size_t>(shape_size), p_type->Size());
void* p_data = nullptr;
if(shape_size > 0)
p_data = allocator->AllocArray(static_cast<size_t>(shape_size), p_type->Size());
Init(p_type, shape, p_data, allocator, offset);
}