Skip to content

Commit 6ee2ffc

Browse files
committed
Fix Accuracy Errors
1 parent 059980f commit 6ee2ffc

1 file changed

Lines changed: 40 additions & 1 deletion

File tree

onnxruntime/core/mlas/lib/sgemm.cpp

Lines changed: 40 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1095,7 +1095,46 @@ float* MlasSgemmKernelLoop(
10951095
size_t ldc,
10961096
float alpha,
10971097
bool ZeroMode
1098-
)
1098+
)
1099+
/*++
1100+
1101+
Routine Description:
1102+
1103+
This routine steps through the rows of the input and output matrices calling
1104+
the kernel until all rows have been processed.
1105+
1106+
Arguments:
1107+
1108+
A - Supplies the address of matrix A.
1109+
1110+
B - Supplies the address of matrix B. The matrix data has been packed using
1111+
MlasSgemmCopyPackB or MlasSgemmTransposePackB.
1112+
1113+
C - Supplies the address of matrix C.
1114+
1115+
CountK - Supplies the number of columns from matrix A and the number of rows
1116+
from matrix B to iterate over.
1117+
1118+
CountM - Supplies the number of rows from matrix A and matrix C to iterate
1119+
over.
1120+
1121+
CountN - Supplies the number of columns from matrix B and matrix C to
1122+
iterate over.
1123+
1124+
lda - Supplies the first dimension of matrix A.
1125+
1126+
ldc - Supplies the first dimension of matrix C.
1127+
1128+
alpha - Supplies the scalar alpha multiplier (see SGEMM definition).
1129+
1130+
ZeroMode - Supplies true if the output matrix must be zero initialized,
1131+
else false if the output matrix is accumulated into.
1132+
1133+
Return Value:
1134+
1135+
Returns the next address of matrix C.
1136+
1137+
--*/
10991138
{
11001139
while (CountM > 0) {
11011140

0 commit comments

Comments
 (0)