File tree Expand file tree Collapse file tree
onnxruntime/core/mlas/lib Expand file tree Collapse file tree Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments