From 9cf706c9ad09abc88d93d69815bfe002da690901 Mon Sep 17 00:00:00 2001
From: yejianwu <yejianwu@xiaomi.com>
Date: Wed, 8 Aug 2018 11:08:07 +0800
Subject: [PATCH] fix include header

---
 mace/core/registry.h    | 1 +
 mace/kernels/proposal.h | 1 +
 2 files changed, 2 insertions(+)

diff --git a/mace/core/registry.h b/mace/core/registry.h
index 277cabb3..ac813287 100644
--- a/mace/core/registry.h
+++ b/mace/core/registry.h
@@ -15,6 +15,7 @@
 #ifndef MACE_CORE_REGISTRY_H_
 #define MACE_CORE_REGISTRY_H_
 
+#include <functional>
 #include <map>
 #include <memory>
 #include <mutex>  // NOLINT(build/c++11)
diff --git a/mace/kernels/proposal.h b/mace/kernels/proposal.h
index 748264e5..89f79b7f 100644
--- a/mace/kernels/proposal.h
+++ b/mace/kernels/proposal.h
@@ -16,6 +16,7 @@
 #define MACE_KERNELS_PROPOSAL_H_
 
 #include <algorithm>
+#include <cmath>
 #include <vector>
 
 #include "mace/core/future.h"
-- 
GitLab