diff --git a/transformer_engine/common/fused_attn/utils.cu b/transformer_engine/common/fused_attn/utils.cu
index c6c8957b1b0..b74dcfe4344 100644
--- a/transformer_engine/common/fused_attn/utils.cu
+++ b/transformer_engine/common/fused_attn/utils.cu
@@ -5,6 +5,7 @@
  ************************************************************************/
 
 #include <algorithm>
+#include <cassert>
 #include <cmath>
 
 #include "../common.h"