diff --git a/mindspore/ccsrc/device/ascend/kernel_build_ascend.cc b/mindspore/ccsrc/device/ascend/kernel_build_ascend.cc
index 1dc688c72f267a0bbe0e09c2f5e5f2d254de5a3f..a2f7a1c697027af2781f88844de64ac3fe853ad7 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"