Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

integrate oneDNN implementation of RNN #1109

Open
jianyizh opened this issue Nov 21, 2024 · 1 comment
Open

integrate oneDNN implementation of RNN #1109

jianyizh opened this issue Nov 21, 2024 · 1 comment
Assignees
@xytintel
Labels
kernel_optimization xpu-op
Milestone
PT2.8

Comments

@jianyizh
Copy link
Contributor

🚀 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
@jianyizh jianyizh mentioned this issue Nov 21, 2024
Open
@jianyizh jianyizh assigned jianyizh and unassigned xytintel Dec 25, 2024
@jianyizh
Copy link
Contributor Author

I will check if we can use thnn_fused_lstm_cell_forward

@jianyizh jianyizh assigned xytintel and unassigned jianyizh Jan 3, 2025
@jianyizh jianyizh changed the title Add RNN related op on XPU integrate oneDNN implementation of RNN Jan 3, 2025
@xytintel xytintel added this to the PT2.8 milestone Jan 3, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@xytintel xytintel

Labels
kernel_optimization xpu-op
Projects
None yet
Milestone
PT2.8
Development

No branches or pull requests
2 participants
@jianyizh @xytintel