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.

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

[Op] Support for repeat_interleave backward #1349

Open
pmarkovicTT opened this issue Feb 28, 2025 · 0 comments
Open

[Op] Support for repeat_interleave backward #1349

pmarkovicTT opened this issue Feb 28, 2025 · 0 comments
Assignees

Comments

@pmarkovicTT
Copy link
Contributor

Problem

Test for Llama 3.2 model backward fails with NotImplementedError: repeat_interleave

Seems like there is missing backward implementation for the op in forge/forge/op/eval/forge/tm.py

Repro

Run pytest forge/test/mlir/llama/test_llama_backward.py with @pytest.mark.parametrize("model_path", ["meta-llama/Llama-3.2-1B"])

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants