map neon allocator

mace/core/allocator.cc

@@ -16,9 +16,11 @@ void SetCPUAllocator(CPUAllocator* alloc) {
 }
 
 Allocator* GetDeviceAllocator(DeviceType type) {
-  if (type == DeviceType::CPU) {
+  switch (type) {
+    case DeviceType::CPU:
+    case DeviceType::NEON:
       return cpu_allocator();
-  } else {
+    default:
       REQUIRE(false, "device type ", type, " is not supported.");
   }
   return nullptr;

mace/core/operator.cc

@@ -23,7 +23,7 @@ MACE_DEFINE_REGISTRY(
     OperatorBase,
     const OperatorDef&,
     Workspace*);
-MACE_REGISTER_DEVICE_TYPE(DeviceType::NEON, CPUOperatorRegistry);
+MACE_REGISTER_DEVICE_TYPE(DeviceType::NEON, NEONOperatorRegistry);
 
 unique_ptr<OperatorBase> CreateOperator(
     const OperatorDef& operator_def,