From 7494a180a42d0c00ec0a43fb996f889f8ebd9a85 Mon Sep 17 00:00:00 2001
From: "Dongxu.Wang" <wangdongxuking61@gmail.com>
Date: Tue, 7 Jan 2025 11:05:12 +0800
Subject: [PATCH] fix bug: arch/mma_sm60.h Mma<2,2,1> calculate wrong (#1989)

---
 include/cutlass/arch/mma_sm60.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/include/cutlass/arch/mma_sm60.h b/include/cutlass/arch/mma_sm60.h
index 3e3c71ef3..32322bc4e 100644
--- a/include/cutlass/arch/mma_sm60.h
+++ b/include/cutlass/arch/mma_sm60.h
@@ -228,7 +228,7 @@ struct Mma<
     __half2 const *C = reinterpret_cast<__half2 const *>(&c);
 
     __half2 Dlo = __hfma2(Alo, B, C[0]);
-    __half2 Dhi = __hfma2(Ahi, B, C[0]);
+    __half2 Dhi = __hfma2(Ahi, B, C[1]);
 
     Array<half_t, 2> * D = reinterpret_cast<Array<half_t, 2> *>(&d);