sbgemm: add missing cblas_sbgemm definition
authorWangyang Guo <wangyang.guo@intel.com>
Thu, 12 Aug 2021 16:51:24 +0000 (00:51 +0800)
committerWangyang Guo <wangyang.guo@intel.com>
Mon, 30 Aug 2021 09:40:30 +0000 (17:40 +0800)
cblas.h
interface/gemm.c

diff --git a/cblas.h b/cblas.h
index f0220eb..a5ad25a 100644 (file)
--- a/cblas.h
+++ b/cblas.h
@@ -400,6 +400,8 @@ void   cblas_dbf16tod(OPENBLAS_CONST blasint n, OPENBLAS_CONST bfloat16 *in, OPE
 float  cblas_sbdot(OPENBLAS_CONST blasint n, OPENBLAS_CONST bfloat16 *x, OPENBLAS_CONST blasint incx, OPENBLAS_CONST bfloat16 *y, OPENBLAS_CONST blasint incy);
 void   cblas_sbgemv(OPENBLAS_CONST enum CBLAS_ORDER order,  OPENBLAS_CONST enum CBLAS_TRANSPOSE trans,  OPENBLAS_CONST blasint m, OPENBLAS_CONST blasint n, OPENBLAS_CONST float alpha, OPENBLAS_CONST bfloat16 *a, OPENBLAS_CONST blasint lda, OPENBLAS_CONST bfloat16 *x, OPENBLAS_CONST blasint incx, OPENBLAS_CONST float beta, float *y, OPENBLAS_CONST blasint incy);
 
+void   cblas_sbgemm(OPENBLAS_CONST enum CBLAS_ORDER Order, OPENBLAS_CONST enum CBLAS_TRANSPOSE TransA, OPENBLAS_CONST enum CBLAS_TRANSPOSE TransB, OPENBLAS_CONST blasint M, OPENBLAS_CONST blasint N, OPENBLAS_CONST blasint K,
+                   OPENBLAS_CONST float alpha, OPENBLAS_CONST bfloat16 *A, OPENBLAS_CONST blasint lda, OPENBLAS_CONST bfloat16 *B, OPENBLAS_CONST blasint ldb, OPENBLAS_CONST float beta, float *C, OPENBLAS_CONST blasint ldc);
 #ifdef __cplusplus
 }
 #endif  /* __cplusplus */
index 47e0ca0..6dcc540 100644 (file)
@@ -273,8 +273,8 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE TransA, enum CBLAS_TRANS
           blasint m, blasint n, blasint k,
 #ifndef COMPLEX
           FLOAT alpha,
-          FLOAT *a, blasint lda,
-          FLOAT *b, blasint ldb,
+          IFLOAT *a, blasint lda,
+          IFLOAT *b, blasint ldb,
           FLOAT beta,
           FLOAT *c, blasint ldc) {
 #else