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 * D = reinterpret_cast *>(&d);