-
Notifications
You must be signed in to change notification settings - Fork 3.8k
Closed
Labels
Description
Feature Request
Recently, TVM has supported the incoming MXNet 2.0 frontend via a series of PRs: #6054, #6699. This enables the new GluonNLP 1.0, which has been upgraded from MXNet 1.x to MXNet 2.0 to convert to TVM graph (as added in dmlc/gluon-nlp#1390).
However, not all operators in MXNet support conversion to TVM. The following is the list of operators that need to be supported.
- _npi_subtract_scalar BART model
- _npi_stack, BART model
- _npi_advanced_indexing_multiple, BART model. This is triggered when we call
a[idx1, idx2]. Also see the MXNet-side implementation.