hgemm: enable CPU hgemm only when MKL is provided.

icl-utk-edu · Dec 14, 2023 · 8678c82 · 8678c82
1 parent 70d893b
commit 8678c82
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/gemm.cc b/src/gemm.cc
@@ -30,10 +30,12 @@ inline void gemm(
     float16 beta,
     float16*       C, blas_int ldc )
 {
+#ifdef BLAS_HAVE_MKL
     BLAS_hgemm( &transA, &transB, &m, &n, &k,
         (MKL_F16*)&alpha,  (MKL_F16*)A, &lda,
                            (MKL_F16*)B, &ldb,
         (MKL_F16*)&beta,   (MKL_F16*)C, &ldc );
+#endif
 }
 
 //------------------------------------------------------------------------------