[Fix] Fix aligned allocations on Android (open-mmlab#1612)

* fix android alignment * fix typo * fix size
irexyc · Jan 11, 2023 · f4b2187 · f4b2187
1 parent 28750ac
commit f4b2187
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/csrc/mmdeploy/device/cpu/cpu_device.cpp b/csrc/mmdeploy/device/cpu/cpu_device.cpp
@@ -11,9 +11,12 @@ class CpuHostMemory : public NonCopyable {
  public:
   CpuHostMemory() : size_(), data_(), owned_data_{false} {}
   Result<void> Init(size_t size, size_t alignment) {
-    size_t space = (size + alignment - 1) / alignment * alignment;
+    alignment = std::max(alignment, sizeof(void*));
+    auto space = (size + alignment - 1) / alignment * alignment;
 #ifdef _MSC_VER
     data_ = _aligned_malloc(space, alignment);
+#elif defined(ANDROID)
+    posix_memalign(&data_, alignment, space);
 #else
     data_ = std::aligned_alloc(alignment, space);
 #endif