tvm
tvm copied to clipboard
[Relax][Runtime] RNNState for Space State Models
This commit adds the RNNState class to the Relax VM, similar to the PagedKVCache, for space state models like RWKV and mamba
cc @tqchen @MasterJH5574
Refactored. Now AttentionKVCache and RNNState have the same base class. would be great if @MasterJH5574 @tqchen Can take a look
please fix the lint error
CI is green now :)