Add missing import for LIMITS in benchmarks

ZzEeKkAa · ZzEeKkAa · commit c6fe75bd6d83 · 2024-10-03T16:41:32.000Z
diff --git a/benchmarks/xetla_kernel/flash_attention/fmha_forward_v5.h b/benchmarks/xetla_kernel/flash_attention/fmha_forward_v5.h
@@ -16,6 +16,8 @@
 #ifndef TRITONBENCHMARK_FMHA_FWD_V5_H
 #define TRITONBENCHMARK_FMHA_FWD_V5_H
 
+#include <cmath>
+
 #include "fmha_policy_v2.h"
 #include "fmha_utils.h"
 #include "xetla.hpp"
