mirror of
https://github.com/ggerganov/whisper.cpp.git
synced 2024-12-21 13:37:47 +00:00
ggml : add alternative cblas_sgemm call
This commit is contained in:
parent
832b4f34c9
commit
50a061b313
15
ggml.c
15
ggml.c
@ -4590,13 +4590,22 @@ void ggml_compute_forward_mul_mat_f16_f32(
|
|||||||
// }
|
// }
|
||||||
//}
|
//}
|
||||||
|
|
||||||
// zT = y * xT
|
|
||||||
{
|
{
|
||||||
|
#if 1
|
||||||
|
// zT = y * xT
|
||||||
cblas_sgemm(CblasRowMajor, CblasNoTrans, CblasTrans,
|
cblas_sgemm(CblasRowMajor, CblasNoTrans, CblasTrans,
|
||||||
ne11, ne01, ne10,
|
ne11, ne01, ne10,
|
||||||
1.0f, y, ne10,
|
1.0f, y, ne00,
|
||||||
x, ne10,
|
x, ne00,
|
||||||
0.0f, d, ne01);
|
0.0f, d, ne01);
|
||||||
|
#else
|
||||||
|
// zT = (xT * y)T
|
||||||
|
cblas_sgemm(CblasColMajor, CblasTrans, CblasNoTrans,
|
||||||
|
ne01, ne11, ne10,
|
||||||
|
1.0f, x, ne00,
|
||||||
|
y, ne00,
|
||||||
|
0.0f, d, ne01);
|
||||||
|
#endif
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user