Skip to content

Commit 3448836

Browse files
authored
Merge pull request #6085 from martin-frbg/smallgemmalpha
Prevent the small matrix kernels from getting called with alpha=0
2 parents 8b0e417 + 5df59af commit 3448836

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

‎interface/gemm.c‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -733,6 +733,7 @@ double _Complex c_be={be[0],be[1]};
733733
FUNCTION_PROFILE_START();
734734

735735
#if USE_SMALL_MATRIX_OPT
736+
if (*(FLOAT*)args.alpha != 0.0) {
736737
#if !defined(COMPLEX)
737738
if(GEMM_SMALL_MATRIX_PERMIT(transa, transb, args.m, args.n, args.k, *(FLOAT *)(args.alpha), *(FLOAT *)(args.beta))){
738739

@@ -753,6 +754,7 @@ double _Complex c_be={be[0],be[1]};
753754
return;
754755
}
755756
#endif
757+
}
756758
#endif
757759

758760
buffer = (XFLOAT *)blas_memory_alloc(0);

0 commit comments

Comments
 (0)