search

microsoft / microxcaling

PyTorch emulation library for Microscaling (MX)-compatible data formats
MIT License
123 stars 14 forks source link

Support for LSTMCell #12

Closed rjfnobre closed 6 months ago

rjfnobre commented 7 months ago

It seems that there is currently no support for torch.nn.LSTMCell.

Is there a simple workaround to adapt code with LSTMCell to enable use the microxscaling library?

rizhao-msft commented 6 months ago

We have an implementation for nn.LSTM that we didn't include in the initial release. Would that be enough? See e825385

rjfnobre commented 6 months ago

Thank you for including the implementation for nn.LSTM. I believe it will be helpful for many people.

You can close this thread.