sxjscience opened a new issue #6721: URL: https://github.com/apache/incubator-tvm/issues/6721
Recently, TVM has supported the incoming MXNet 2.0 frontend via a series of PRs: https://github.com/apache/incubator-tvm/pull/6054, https://github.com/apache/incubator-tvm/pull/6699. This enables the new GluonNLP 1.0, which has upgraded from MXNet 1.x to MXNet 2.0 to convert to TVM graph (as added in https://github.com/dmlc/gluon-nlp/pull/1390). However, I noticed that there are three missing operators when I profile against the [BART model](https://github.com/dmlc/gluon-nlp/blob/master/src/gluonnlp/models/bart.py). - _npi_subtract_scalar - _npi_stack - _npi_advanced_indexing_multiple The first two operators are pretty straightforward to support while the `_npi_advanced_indexing_multiple` is more complicated and is triggered when we call `a[idx1, idx2]`. The MXNet-side implementation of `_npi_advanced_indexing_multiple` is here: https://github.com/apache/incubator-mxnet/blob/6bbd53107aa16fc41e8d462cf5dc46fb70d592df/src/operator/numpy/np_indexing_op.cc#L479-L491 Since I do not have the bandwidth in supporting this operator. Help is really appreciated. ---------------------------------------------------------------- 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]
