fix cpu aligned_alloc size (#8569)

Signed-off-by: daquexian <daquexian566@gmail.com> Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
Oneflow-Inc · Jul 6, 2022 · 11e6278 · 11e6278
1 parent 51b9a1e
commit 11e6278
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/oneflow/core/ep/cpu/cpu_device.cpp b/oneflow/core/ep/cpu/cpu_device.cpp
@@ -44,7 +44,7 @@ Maybe<void> CpuDevice::Alloc(const AllocationOptions& options, void** ptr, size_
     CHECK_OR_RETURN(device);
     return device->AllocPinned(options, ptr, size);
   } else {
-    *ptr = aligned_alloc(kMaxAlignmentRequirement, size);
+    *ptr = aligned_alloc(kMaxAlignmentRequirement, RoundUp(size, kMaxAlignmentRequirement));
     if (*ptr == nullptr) {
       return Error::RuntimeError() << "allocate failed";
     } else {