diff options
Diffstat (limited to 'src/runtime/TensorAllocator.cpp')
-rw-r--r-- | src/runtime/TensorAllocator.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/runtime/TensorAllocator.cpp b/src/runtime/TensorAllocator.cpp index 993a95b6c3..c84a2719d8 100644 --- a/src/runtime/TensorAllocator.cpp +++ b/src/runtime/TensorAllocator.cpp @@ -138,7 +138,7 @@ void TensorAllocator::allocate() if(_associated_memory_group == nullptr) { - _memory = Memory(std::make_shared<MemoryRegion>(info().total_size())); + _memory = Memory(std::make_shared<MemoryRegion>(info().total_size(), alignment())); } else { |