matteosal commented on issue #20301:
URL:
https://github.com/apache/incubator-mxnet/issues/20301#issuecomment-872358047
> @matteosal use MKL_VERBOSE=1 when you're running your reproduction -
> I'm suspecting that something is wrong with your mkl installation. Can you
try using mkl from conda env?
> `conda install mkl mkl-inlcude mkl-service -c intel`
I've rebuilt linking to this MKL (which should be version 2021.2) and still
get the same crash.
I have also tried building with `-DUSE_MKL_LAYERNORM=ON` (and my version of
MKL) and this script doesn't crash:
```
import mxnet as mx
sym = mx.sym.LayerNorm(mx.sym.Variable('data'), mx.sym.Variable('gamma'),
mx.sym.Variable('beta'))
ex = sym._bind(
mx.cpu(),
{'data': mx.nd.ones((2, 3)), 'gamma': mx.nd.ones((3)), 'beta':
mx.nd.ones((3))}
)
ex.forward()
print('done')
```
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: issues-unsubscr...@mxnet.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscr...@mxnet.apache.org
For additional commands, e-mail: issues-h...@mxnet.apache.org
