Allow opt into GEMM -> GEMV forwarding

2024-07-24 11:25:28 +01:00
parent 90eb863d4b
commit b26424c6a2
3 changed files with 17 additions and 1 deletions
--- a/interface/gemm.c
+++ b/interface/gemm.c
@@ -498,7 +498,7 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE TransA, enum CBLAS_TRANS
 	 args.m, args.n, args.k, args.lda, args.ldb, args.ldc);
 #endif

-#if !defined(GEMM3M) && !defined(COMPLEX)
+#if defined(GEMM_GEMV_FORWARD) && !defined(GEMM3M) && !defined(COMPLEX)
  // Check if we can convert GEMM -> GEMV
  if (args.k != 0) {
    if (args.n == 1) {