You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by GitBox <gi...@apache.org> on 2020/08/26 17:55:06 UTC

[GitHub] [incubator-mxnet] Zha0q1 opened a new issue #19020: cub apis do not support large tensor

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:
users@infra.apache.org

[GitHub] [incubator-mxnet] Zha0q1 edited a comment on issue #19020: cub apis do not support large tensor

Posted by GitBox <gi...@apache.org>.
Zha0q1 edited a comment on issue #19020:
URL: https://github.com/apache/incubator-mxnet/issues/19020#issuecomment-682092534


   @sandeep-krishnamurthy @access2rohit For your reference 


----------------------------------------------------------------
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:
users@infra.apache.org

[GitHub] [incubator-mxnet] Zha0q1 commented on issue #19020: cub apis do not support large tensor

Posted by GitBox <gi...@apache.org>.
Zha0q1 commented on issue #19020:
URL: https://github.com/apache/incubator-mxnet/issues/19020#issuecomment-682092534


   @sandeep-krishnamurthy @access2rohit For your guys' reference 


----------------------------------------------------------------
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:
users@infra.apache.org