diff --git a/mindspore/ccsrc/device/ascend/kernel_build_ascend.cc b/mindspore/ccsrc/device/ascend/kernel_build_ascend.cc
index 1dc688c72f2..a2f7a1c6970 100644
--- a/mindspore/ccsrc/device/ascend/kernel_build_ascend.cc
+++ b/mindspore/ccsrc/device/ascend/kernel_build_ascend.cc
@@ -19,6 +19,8 @@
 #include <vector>
 #include <string>
 #include <memory>
+#include <set>
+#include <map>
 #include <functional>
 
 #include "device/ascend/kernel_select_ascend.h"