[GitHub] sbodenstein edited a comment on issue #14124: SequenceMask is slow on GPU for BERT

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

sbodenstein edited a comment on issue #14124: SequenceMask is slow on GPU for BERT
URL: https://github.com/apache/incubator-mxnet/issues/14124#issuecomment-463993614
 
 
   @eric-haibin-lin: that is a great catch, this should definitely be fast enough to throttle BERT.
   
   I won't have time in the next 2 weeks to work on this. So would be happy for someone else to rewrite if this is urgent! 
   
   Its also a pity that neither TensorFlow nor PyTorch have this operator (and an efficient GPU implementation we can quickly adapt).

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