From 2f9ebe53a859c1843ba71cdcb6e47fd06689a234 Mon Sep 17 00:00:00 2001 From: Natalia Gimelshein Date: Sun, 19 Jul 2020 10:23:29 -0700 Subject: [PATCH] disable mkl for expm1 --- aten/src/ATen/cpu/vml.h | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/aten/src/ATen/cpu/vml.h b/aten/src/ATen/cpu/vml.h index 8e20605f6792f..0670465f566cc 100644 --- a/aten/src/ATen/cpu/vml.h +++ b/aten/src/ATen/cpu/vml.h @@ -183,8 +183,8 @@ static_assert( IMPLEMENT_VML_MKL_STUB(op, mklop, float, s) \ IMPLEMENT_VML_MKL_STUB(op, mklop, double, d) -// NB: abs, cosh and sinh were temporarily disabled due to issues with Apple clang - +// NB: abs, cosh and sinh were temporarily disabled due to issues with Apple +// NB: expm1 is disabled because on some configs it produces expm1(nan)=-1 IMPLEMENT_VML_MKL(abs, Abs) IMPLEMENT_VML_MKL(acos, Acos) IMPLEMENT_VML_MKL(asin, Asin) @@ -195,7 +195,7 @@ IMPLEMENT_VML_MKL(erf, Erf) IMPLEMENT_VML_MKL(erfc, Erfc) IMPLEMENT_VML_MKL(erfinv, ErfInv) IMPLEMENT_VML_MKL(exp, Exp) -IMPLEMENT_VML_MKL(expm1, Expm1) +// IMPLEMENT_VML_MKL(expm1, Expm1) IMPLEMENT_VML_MKL(log, Ln) IMPLEMENT_VML_MKL(log10, Log10) IMPLEMENT_VML_MKL(log1p, Log1p)