ggml : add alternative cblas_sgemm call

pull/245/head
Georgi Gerganov 2 years ago
parent 832b4f34c9
commit 50a061b313
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735

@ -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,
ne11, ne01, ne10,
1.0f, y, ne10,
x, ne10,
1.0f, y, ne00,
x, ne00,
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…
Cancel
Save