It seems that TVM duplicates parameters in LSTMs and GRUs for each step in the input series.
This behavior has occurred before in the past but was fixed.
Maybe this issue is related to the refactoring done in #12213 but I could not really locate the source of the error.
I use TVM on commit ecbe4ca
I attached a zip file containing
- a toy lstm model that takes an input with two steps.
- a sample script to generate C code for it with TVM
- the generated code.
If you look into module_0.txt you can clearly see that identical parameters are in there twice.
@zhang-yi-chi @mbrookhart @AndrewZhaoLuo @vdkhoi @UlrikHjort @MichaelJKlaiber
lstm_bug.zip
r