commit 1dadbf71d9e2468dad817bcb66e0b6bd7279b980
author Marat Dukhan <maratek@google.com> Tue Oct 01 10:46:20 2019 -0700
committer XNNPACK Team <xnnpack-github-robot@google.com> Tue Oct 01 10:46:41 2019 -0700
tree 89d71950b3a0eb6920da8a317afd39b96ea7aa32
parent 629a33eef8ef2fa86764a16da85a4770f3898953

Limit direct dependencies on cpuinfo

PiperOrigin-RevId: 272245408

src/xnnpack/allocator.h

diff --git a/src/xnnpack/allocator.h b/src/xnnpack/allocator.h
index 303aa37..c946656 100644
--- a/src/xnnpack/allocator.h
+++ b/src/xnnpack/allocator.h
@@ -12,7 +12,7 @@
   #include <malloc.h>
 #endif
 
-#include <cpuinfo.h>
+#include <xnnpack/common.h>
 
 extern int posix_memalign(void **memptr, size_t alignment, size_t size);
 
@@ -22,7 +22,7 @@
 
 inline static void* xnn_allocate_memory(size_t memory_size) {
   void* memory_ptr = NULL;
-#if CPUINFO_ARCH_ASMJS || CPUINFO_ARCH_WASM
+#if XNN_ARCH_ASMJS || XNN_ARCH_WASM
   memory_ptr = malloc(memory_size);
 #elif defined(__ANDROID__)
   memory_ptr = memalign(XNN_ALLOCATION_ALIGNMENT, memory_size);