Strange issue when using HybridBlock in a symbolic program


#1

I defined a custom operation as a HybridBlock and added it to a symbolic codebase. However, when I try to load the model, I get the following error:

MXNetError: Failed loading Op _mul4 of type elemwise_mul: [05:31:13] src/core/op.cc:55: Check failed: op != nullptr Operator elemwise_mul is not registered

Stack trace returned 10 entries:
[bt] (0) /usr/local/lib/python3.5/dist-packages/mxnet/libmxnet.so(+0x2d55679) [0x7f48d2640679]
[bt] (1) /usr/local/lib/python3.5/dist-packages/mxnet/libmxnet.so(+0x2d840b4) [0x7f48d266f0b4]
[bt] (2) /usr/local/lib/python3.5/dist-packages/mxnet/libmxnet.so(+0x2d88d30) [0x7f48d2673d30]
[bt] (3) /usr/local/lib/python3.5/dist-packages/mxnet/libmxnet.so(+0x2d82ec6) [0x7f48d266dec6]
[bt] (4) /usr/local/lib/python3.5/dist-packages/mxnet/libmxnet.so(+0xf9aa1f) [0x7f48d0885a1f]
[bt] (5) /usr/local/lib/python3.5/dist-packages/mxnet/libmxnet.so(+0x2d58f78) [0x7f48d2643f78]
[bt] (6) /usr/local/lib/python3.5/dist-packages/mxnet/libmxnet.so(+0xf94ead) [0x7f48d087fead]
[bt] (7) /usr/local/lib/python3.5/dist-packages/mxnet/libmxnet.so(+0xf9aa1f) [0x7f48d0885a1f]
[bt] (8) /usr/local/lib/python3.5/dist-packages/mxnet/libmxnet.so(+0x2d58f78) [0x7f48d2643f78]
[bt] (9) /usr/local/lib/python3.5/dist-packages/mxnet/libmxnet.so(+0xe0fcae) [0x7f48d06facae]

Not sure if it’s a bug or it’s the way I implemented the entire thing


#2

Based on CUDA: Unspecified launch failure , my guess is that I didn’t do something too wrong?


#3

Can you post a short code snippet?


#4

Code Snippet is here (Example of using a Gluon HybridBlock in a Symbol Program)


#5

Looks like upgrading to master solves this issue. My guess is elementwise_mul wasn’t registered till recently?


#6

Yes, that’s the root cause.