larroy commented on issue #14522: mx.nd.Custom conflicts with memory management URL: https://github.com/apache/incubator-mxnet/issues/14522#issuecomment-478154675 This repro in imperative mode isolates from the exception handling problem in custom ops: ``` import mxnet as mx n = 100 a = mx.nd.random.uniform(shape=(n, 6000, 1)) b = mx.nd.random.uniform(shape=(n, 1, 7000)) c = mx.nd.batch_dot(a, b) c.wait_to_read() print(c.shape) ```
---------------------------------------------------------------- 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. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services