diff --git a/paddle/phi/core/distributed/auto_parallel/utils.h b/paddle/phi/core/distributed/auto_parallel/utils.h
index 63036c9b7e93a9964c937d358d47ea711ca6e224..c9e69dd550abb89c0cca9f4b9fba249ac2cb6a1f 100644
--- a/paddle/phi/core/distributed/auto_parallel/utils.h
+++ b/paddle/phi/core/distributed/auto_parallel/utils.h
@@ -14,6 +14,7 @@ limitations under the License. */
 #pragma once
 
 #include <algorithm>
+#include <map>
 #include <sstream>
 #include <string>
 #include <unordered_map>