cc @gussmith23
I’m trying to import MLPerf’s RNN-T model into TVM through the Relay frontend. I’m facing similar issues to this thread, but the errors are different!
I was able to get a trace of the model, but after a call to relay.frontend.from_pytorch
, I ran into a NotImplemented Error
saying that The following operators are not implemented: ['aten::copy_']
.
I was able to get aten::copy_
implemented by following the instructions in this thread, yielding an additional error: File "/root/tvm/python/tvm/relay/frontend/pytorch.py", line 2670, in lstm_layers raise NotImplementedError("Dropout for LSTM has not been supported yet!")
.
By uncommenting the two lines of implementation above line 2670, we were able to get past this error, finally reaching a new error on line 3768: NotImplementedError: Unsupported type: ListType
.
I’d appreciate any advice for dealing with this new error!