intel / torch-xpu-ops

Apache License 2.0
30 stars 21 forks source link

Add RNN related op on XPU #1109

Open jianyizh opened 14 hours ago

jianyizh commented 14 hours ago

🚀 The feature, motivation and pitch

There are some models in torch benchmarks using lstm, rnn... CUDA uses cuDNN for these ops. Can we add them using oneDNN?

Alternatives

No response

Additional context

No response