Skip to content

Create a new tensor arg for stride_per_key_per_rank to facilitate torch.export #2950

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

Open
jd7-tr wants to merge 2 commits into
base: main
Choose a base branch
from

Create a new tensor arg for stride_per_key_per_rank to facilitate tor…

96ee8d5
Select commit
Loading
Failed to load commit list.
Open

Create a new tensor arg for stride_per_key_per_rank to facilitate torch.export #2950

Create a new tensor arg for stride_per_key_per_rank to facilitate tor…
96ee8d5
Select commit
Loading
Failed to load commit list.
Facebook GitHub Tools / Meta Import Checks succeeded May 7, 2025 in 0s

All checks passed, this Pull Request can be imported!

This check validates that the Pull Request can be imported to Meta internal infrastructure

View more details on Facebook GitHub Tools