diff --git a/paddle/fluid/framework/selected_rows.h b/paddle/fluid/framework/selected_rows.h
index 24c8b64b0318dedda1f32cb43aeb32913e86513d..1990cb1434a3db4cbb47f554d8c7e69d3b88a226 100644
--- a/paddle/fluid/framework/selected_rows.h
+++ b/paddle/fluid/framework/selected_rows.h
@@ -16,6 +16,7 @@ limitations under the License. */
 
 #include <algorithm>
 #include <memory>
+#include <mutex>  // NOLINT
 #include <utility>
 #include <vector>