diff --git a/bench/banded_attn_proto.cpp b/bench/banded_attn_proto.cpp
index 50e9e02..72b7eb1 100644
--- a/bench/banded_attn_proto.cpp
+++ b/bench/banded_attn_proto.cpp
@@ -9,7 +9,7 @@
 #include <ggml.h>
 #include <ggml-cpu.h>
 #include <ggml-backend.h>
-
+#include <algorithm>
 #include <cmath>
 #include <cstdio>
 #include <cstdlib>