youngkingdom/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix bug: arch/mma_sm60.h Mma<2,2,1> calculate wrong (#1989)

Browse Source
This commit is contained in:
Dongxu.Wang
2025-01-07 11:05:12 +08:00
committed by GitHub
parent cffd5d32b7
commit 7494a180a4
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/cutlass/arch/mma_sm60.h
View File

@ -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);