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


With regards,
Apache Git Services

Reply via email to