diff options
Diffstat (limited to 'src/runtime/cpu/utils/CpuAuxTensorHandler.h')
| -rw-r--r-- | src/runtime/cpu/utils/CpuAuxTensorHandler.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/runtime/cpu/utils/CpuAuxTensorHandler.h b/src/runtime/cpu/utils/CpuAuxTensorHandler.h index 644018a718..0d1c927b5a 100644 --- a/src/runtime/cpu/utils/CpuAuxTensorHandler.h +++ b/src/runtime/cpu/utils/CpuAuxTensorHandler.h @@ -41,6 +41,10 @@ public: CpuAuxTensorHandler(int slot_id, TensorInfo &info, ITensorPack &pack, bool pack_inject = false) : _tensor() { + if(info.total_size() == 0) + { + return; + } _tensor.allocator()->soft_init(info); ITensor *packed_tensor = utils::cast::polymorphic_downcast<ITensor *>(pack.get_tensor(slot_id)); |