diff --git a/cpp/kernels/contextAttentionKernels/contextFMHARunner.cpp b/cpp/kernels/contextAttentionKernels/contextFMHARunner.cpp
index 186e3498..6978d19c 100644
--- a/cpp/kernels/contextAttentionKernels/contextFMHARunner.cpp
+++ b/cpp/kernels/contextAttentionKernels/contextFMHARunner.cpp
@@ -24,7 +24,7 @@
 #include <cstring>
 #include <cuda.h>
 #include <cuda_fp16.h>
-#include <math.h>
+#include <cmath>
 #include <memory>
 #include <mutex>
 #include <unordered_map>
diff --git a/cpp/kernels/contextAttentionKernels/fmhaParams_v2.h b/cpp/kernels/contextAttentionKernels/fmhaParams_v2.h
index 4252197d..72a26bdb 100644
--- a/cpp/kernels/contextAttentionKernels/fmhaParams_v2.h
+++ b/cpp/kernels/contextAttentionKernels/fmhaParams_v2.h
@@ -18,7 +18,7 @@
 #pragma once
 
 #include <limits.h>
-#include <math.h>
+#include <cmath>
 #include <stdint.h>
 
 //! \brief Parameters for ALiBi (Attention with Linear Biases) positional encoding