Browse Source

fix(mkl): fix windows mkl LOG compute exception

GitOrigin-RevId: cd2ebaaec1
release-1.1
Megvii Engine Team 4 years ago
parent
commit
1f8e40753f
1 changed files with 5 additions and 1 deletions
  1. +5
    -1
      dnn/src/x86/elemwise/opr_impl.cpp

+ 5
- 1
dnn/src/x86/elemwise/opr_impl.cpp View File

@@ -153,7 +153,11 @@ bool ElemwiseImpl::exec_unary() {
auto mkl_dispatch = [&]() {
switch (param().mode) {
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(SIN, vsSin);
DISPATCH_MKL(TANH, vsTanh);


Loading…
Cancel
Save