提交 1f8e4075 编写于 作者: M Megvii Engine Team

fix(mkl): fix windows mkl LOG compute exception

GitOrigin-RevId: cd2ebaaec17e143cf496e1888a5d33a2b936210e
上级 fea5cf62
...@@ -153,7 +153,11 @@ bool ElemwiseImpl::exec_unary() { ...@@ -153,7 +153,11 @@ bool ElemwiseImpl::exec_unary() {
auto mkl_dispatch = [&]() { auto mkl_dispatch = [&]() {
switch (param().mode) { switch (param().mode) {
DISPATCH_MKL(ABS, vsAbs); DISPATCH_MKL(ABS, vsAbs);
DISPATCH_MKL(LOG, vsLn); //! Delete the calculation support of MKL LOG because it will
//! cause VML_STATUS_SING error, the reason is At least one of
//! the input array values causes a divide-by-zero exception or
//! produces an invalid (QNaN) result.
// DISPATCH_MKL(LOG, vsLn);
DISPATCH_MKL(COS, vsCos); DISPATCH_MKL(COS, vsCos);
DISPATCH_MKL(SIN, vsSin); DISPATCH_MKL(SIN, vsSin);
DISPATCH_MKL(TANH, vsTanh); DISPATCH_MKL(TANH, vsTanh);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册