Fix doc

codeplaysoftware · Aug 2, 2023 · c231c90 · c231c90
1 parent 831ab63
commit c231c90
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/doc/Gemm.md b/doc/Gemm.md
@@ -249,15 +249,15 @@ namespace internal {
 // gemm
 template typename SB_Handle::event_t _gemm(
     SB_Handle& sb_handle, char _TransA, char _TransB, ${INDEX_TYPE} _M,
-    ${INDEX_TYPE} _N, ${INDEX_TYPE} _K, ${DATA_TYPE} _alpha, ${DATA_TYPE} a_,
-    ${INDEX_TYPE} _lda, ${DATA_TYPE} b_, ${INDEX_TYPE} _ldb,
-    ${DATA_TYPE} _beta, ${DATA_TYPE} _C, ${INDEX_TYPE} _ldc);
+    ${INDEX_TYPE} _N, ${INDEX_TYPE} _K, ${DATA_TYPE} _alpha, BufferIterator<${DATA_TYPE}> a_,
+    ${INDEX_TYPE} _lda, BufferIterator<${DATA_TYPE}> b_, ${INDEX_TYPE} _ldb,
+    ${DATA_TYPE} _beta, BufferIterator<${DATA_TYPE}> _C, ${INDEX_TYPE} _ldc);
 // batched gemm
 template typename SB_Handle::event_t _gemm_batched(
     SB_Handle& sb_handle, char _TransA, char _TransB, ${INDEX_TYPE} _M,
-    ${INDEX_TYPE} _N, ${INDEX_TYPE} _K, ${DATA_TYPE} _alpha, ${DATA_TYPE} a_,
-    ${INDEX_TYPE} _lda, ${INDEX_TYPE} _stridea, ${DATA_TYPE} b_, ${INDEX_TYPE} _ldb, 
-    ${INDEX_TYPE} _strideb, ${DATA_TYPE} _beta, ${DATA_TYPE} _C, ${INDEX_TYPE} _ldc,
+    ${INDEX_TYPE} _N, ${INDEX_TYPE} _K, ${DATA_TYPE} _alpha, BufferIterator<${DATA_TYPE}> a_,
+    ${INDEX_TYPE} _lda, ${INDEX_TYPE} _stridea, BufferIterator<${DATA_TYPE}> b_, ${INDEX_TYPE} _ldb, 
+    ${INDEX_TYPE} _strideb, ${DATA_TYPE} _beta, BufferIterator<${DATA_TYPE}> _C, ${INDEX_TYPE} _ldc,
     ${INDEX_TYPE} _stridec, ${INDEX_TYPE} batch_size, gemm_batch_type_t batch_type);
 }  // namespace internal
 }  // namespace blas