[Runtime] Make runtime compatible with android ndk api 15 (#2446)

apache · Jan 17, 2019 · 52e3bd3 · 52e3bd3
1 parent 6783d37
commit 52e3bd3
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/runtime/cpu_device_api.cc b/src/runtime/cpu_device_api.cc
@@ -10,6 +10,10 @@
 #include <cstring>
 #include "workspace_pool.h"
 
+#ifdef __ANDROID__
+#include <android/api-level.h>
+#endif
+
 namespace tvm {
 namespace runtime {
 class CPUDeviceAPI final : public DeviceAPI {
@@ -28,10 +32,11 @@ class CPUDeviceAPI final : public DeviceAPI {
 #if _MSC_VER
     ptr = _aligned_malloc(nbytes, alignment);
     if (ptr == nullptr) throw std::bad_alloc();
-#elif defined(_LIBCPP_SGX_CONFIG)
+#elif defined(_LIBCPP_SGX_CONFIG) || (defined(__ANDROID__) && __ANDROID_API__ < 16)
     ptr = memalign(alignment, nbytes);
     if (ptr == nullptr) throw std::bad_alloc();
 #else
+    // posix_memalign is available in android ndk since __ANDROID_API__ >= 16
     int ret = posix_memalign(&ptr, alignment, nbytes);
     if (ret != 0) throw std::bad_alloc();
 #endif