From 9e82e8fd15b9e960dca3e34ada0cd63b165334b7 Mon Sep 17 00:00:00 2001
From: "Alex Y. Chan" <alechan@nvidia.com>
Date: Fri, 11 Sep 2026 19:07:42 +0100
Subject: [PATCH 1/2] Insert assert header

Signed-off-by: Alex Y. Chan <alechan@nvidia.com>
---
 transformer_engine/common/fused_attn/utils.cu | 1 +
 1 file changed, 1 insertion(+)

diff --git a/transformer_engine/common/fused_attn/utils.cu b/transformer_engine/common/fused_attn/utils.cu
index c6c8957b1b..9f13b249f6 100644
--- a/transformer_engine/common/fused_attn/utils.cu
+++ b/transformer_engine/common/fused_attn/utils.cu
@@ -6,6 +6,7 @@
 
 #include <algorithm>
 #include <cmath>
+#include <cassert>
 
 #include "../common.h"
 #include "../util/cuda_runtime.h"

From a632a176a28ccc01eead6266913f61dab589dba3 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Fri, 11 Sep 2026 18:08:30 +0000
Subject: [PATCH 2/2] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 transformer_engine/common/fused_attn/utils.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/transformer_engine/common/fused_attn/utils.cu b/transformer_engine/common/fused_attn/utils.cu
index 9f13b249f6..b74dcfe434 100644
--- a/transformer_engine/common/fused_attn/utils.cu
+++ b/transformer_engine/common/fused_attn/utils.cu
@@ -5,8 +5,8 @@
  ************************************************************************/
 
 #include <algorithm>
-#include <cmath>
 #include <cassert>
+#include <cmath>
 
 #include "../common.h"
 #include "../util/cuda_runtime.h"