[GitHub] eric-haibin-lin commented on issue #11241: Conv1D throws CUDNN_STATUS_EXECUTION_FAILED

[GitBox <gi...@apache.org>]

eric-haibin-lin commented on issue #11241: Conv1D throws CUDNN_STATUS_EXECUTION_FAILED
URL: https://github.com/apache/incubator-mxnet/issues/11241#issuecomment-396484475
 
 
   Can be also reproduced by the following code:
   ```
   import mxnet as mx
   from mxnet import nd, sym, autograd
   from mxnet.gluon import nn, HybridBlock, Block
   
   if __name__ == '__main__':
       ctx = mx.gpu()
       x = mx.nd.ones((1L, 65536L, 1560L), ctx=ctx)
       net = nn.Conv1D(channels=256, kernel_size=2, layout='NCW', use_bias=False)
       net.initialize(ctx=ctx)
   
       for p in net.collect_params().values():
           p.grad_req = 'add'
   
       with autograd.record():
           y = net(x)
       y.backward()
       print(net.weight.grad())
   ```
   
   with `pip install mxnet-cu90 --pre`

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services