DNXie opened a new issue, #21017: URL: https://github.com/apache/incubator-mxnet/issues/21017
## Description mxnet.gluon.contrib.nn.PixelShuffle1D crashes with a segmentation fault. This is also reproducible in mxnet nightly version. [Gist](https://colab.research.google.com/drive/1GO_rYa8QXTx3pFWFPvfiUDEfx45OYyA5?usp=sharing). ### Reproduce and Error Message ``` import numpy as np import mxnet layer = mxnet.gluon.contrib.nn.PixelShuffle1D(factor=1) layer.initialize() model_input=mxnet.nd.ones((10,10,10,10)) pred = layer(model_input) print(np.isnan(pred).any()) ``` Error Message: ``` Segmentation fault: 11 Segmentation fault: 11 Segmentation fault: 11 Segmentation fault: 11 Segmentation fault: 11 Segmentation fault: 11 Stack trace: [bt] (0) /root/miniconda3/lib/python3.7/site-packages/mxnet/libmxnet.so(+0x3d05b90) [0x7fe5c0ae5b90] [bt] (1) /lib/x86_64-linux-gnu/libc.so.6(+0x3efd0) [0x7fe5d0f5dfd0] [bt] (2) /root/miniconda3/lib/python3.7/site-packages/mkl/../../../libiomp5.so(+0xad378) [0x7fe5cf70f378] [bt] (3) /root/miniconda3/lib/python3.7/site-packages/mkl/../../../libiomp5.so(__kmpc_bound_num_threads+0xa) [0x7fe5cf6d830a] [bt] (4) /root/miniconda3/lib/python3.7/site-packages/mxnet/libmkldnn.so.1(+0x3d5706) [0x7fe5bbed0706] [bt] (5) /root/miniconda3/lib/python3.7/site-packages/mkl/../../../libiomp5.so(+0x9ba2c) [0x7fe5cf6fda2c] [bt] (6) /root/miniconda3/lib/python3.7/site-packages/mkl/../../../libiomp5.so(__kmp_invoke_microtask+0x93) [0x7fe5cf78ecc3] [bt] (7) /root/miniconda3/lib/python3.7/site-packages/mkl/../../../libiomp5.so(+0xb2283) [0x7fe5cf714283] [bt] (8) /root/miniconda3/lib/python3.7/site-packages/mkl/../../../libiomp5.so(+0xb124e) [0x7fe5cf71324e] Segmentation fault (core dumped) ``` Input 2 (this one causes abortion): ``` import numpy as np import mxnet layer = mxnet.gluon.contrib.nn.PixelShuffle1D(factor=1) layer.initialize() model_input = mxnet.nd.ones((10,2,2,2)) pred = layer(model_input) ``` Error message: ``` Segmentation fault: 11 Stack trace: [bt] (0) /root/miniconda3/lib/python3.7/site-packages/mxnet/libmxnet.so(+0x3d05b90) [0x7f1f8b302b90] [bt] (1) /lib/x86_64-linux-gnu/libc.so.6(+0x3efd0) [0x7f1f9b77afd0] [bt] (2) /root/miniconda3/lib/python3.7/site-packages/mxnet/libmxnet.so(mxnet::NDArray::Chunk::SetMKLMem(mxnet::TShape const&, int)+0x40) [0x7f1f8acaaa70] [bt] (3) /root/miniconda3/lib/python3.7/site-packages/mxnet/libmxnet.so(mxnet::NDArray::GetMKLDNNData() const+0x70) [0x7f1f8acaf4a0] [bt] (4) /root/miniconda3/lib/python3.7/site-packages/mxnet/libmxnet.so(mxnet::op::MKLDNNTransposeForward::MKLDNNTransposeForward(mxnet::op::TransposeParam const&, mxnet::NDArray const&)+0x32b) [0x7f1f8799ebbb] [bt] (5) /root/miniconda3/lib/python3.7/site-packages/mxnet/libmxnet.so(mxnet::op::MKLDNNTransposeForward(nnvm::NodeAttrs const&, mxnet::OpContext const&, mxnet::NDArray const&, mxnet::OpReqType const&, mxnet::NDArray const&)+0x5f4) [0x7f1f879a0a34] [bt] (6) /root/miniconda3/lib/python3.7/site-packages/mxnet/libmxnet.so(+0x32570c4) [0x7f1f8a8540c4] [bt] (7) /root/miniconda3/lib/python3.7/site-packages/mxnet/libmxnet.so(mxnet::imperative::PushFComputeEx(std::function<void (nnvm::NodeAttrs const&, mxnet::OpContext const&, std::vector<mxnet::NDArray, std::allocator<mxnet::NDArray> > const&, std::vector<mxnet::OpReqType, std::allocator<mxnet::OpReqType> > const&, std::vector<mxnet::NDArray, std::allocator<mxnet::NDArray> > const&)> const&, nnvm::Op const*, nnvm::NodeAttrs const&, mxnet::Context const&, std::vector<mxnet::engine::Var*, std::allocator<mxnet::engine::Var*> > const&, std::vector<mxnet::engine::Var*, std::allocator<mxnet::engine::Var*> > const&, std::vector<mxnet::Resource, std::allocator<mxnet::Resource> > const&, std::vector<mxnet::NDArray*, std::allocator<mxnet::NDArray*> > const&, std::vector<mxnet::NDArray*, std::allocator<mxnet::NDArray*> > const&, std::vector<mxnet::OpReqType, std::allocator<mxnet::OpReqType> > const&)::{lambda(mxnet::RunContext)#1}::operator()(mxnet::RunContext) const+0x3b4) [0x7f1f8ab7c674 ] [bt] (8) /root/miniconda3/lib/python3.7/site-packages/mxnet/libmxnet.so(+0x34bcbf4) [0x7f1f8aab9bf4] terminate called without an active exception Aborted (core dumped) ``` ## Environment OS: ubuntu 18.04 Python: 3.7.6 pip: 20.0.2 numpy: 1.18.5 mxnet: 1.6.0 -- 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: [email protected] For queries about this service, please contact Infrastructure at: [email protected] --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
