Zha0q1 opened a new issue #19020: URL: https://github.com/apache/incubator-mxnet/issues/19020
I was working on https://github.com/apache/incubator-mxnet/pull/19015 and have fixed the cpu version of the op. However the fix for the gpu version is blocked by some cub apis which use int in the function declaration. One example is: https://github.com/NVlabs/cub/blob/1.8.0/cub/device/device_scan.cuh#L328 This is used here in the nonzero op:https://github.com/apache/incubator-mxnet/blob/master/src/operator/numpy/np_nonzero_op.cu#L71 and also a dozen of other operators. ---------------------------------------------------------------- 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: [email protected]
