diff --git a/mlx/compile_impl.h b/mlx/compile_impl.h
index f76cc463f..56bb02003 100644
--- a/mlx/compile_impl.h
+++ b/mlx/compile_impl.h
@@ -2,6 +2,8 @@
 
 #pragma once
 
+#include <unordered_map>
+
 #include "mlx/array.h"
 
 namespace mlx::core::detail {