[GitHub] sbodenstein opened a new issue #11966: ArgSort binds to a float64 array but forward fails

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

sbodenstein opened a new issue #11966: ArgSort binds to a float64 array but forward fails
URL: https://github.com/apache/incubator-mxnet/issues/11966
 
 
   `ArgSort` should work on any type, or at least fail during binding if an invalid type is given. But:
   ```
   import numpy as np
   import mxnet as mx
   
   t = "float64"
   s=mx.sym.argsort(mx.sym.Variable("data", dtype=t))
   d=mx.nd.ones((1,2,6,6), dtype=t)
   e=s.bind(mx.cpu(), {'data': d})
   
   <mxnet.executor.Executor at 0x1a13610a20>
   ```
   But then:
   ```
   e.forward()
   
   MXNetError: [16:25:29] include/mxnet/./tensor_blob.h:203: Check failed: mshadow::DataType<DType>::kFlag == type_flag_ TBlob.get_with_shape: data type do not match specified type.Expected: 1 v.s. given 0
   ```
   This works if `t="float32"`.

----------------------------------------------------------------
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