Movatterモバイル変換


[0]ホーム

URL:


Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Sign up
Appearance settings

flex_attention ver.#192

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 ourterms of service andprivacy statement. We’ll occasionally send you account related emails.

Already on GitHub?Sign in to your account

Open
joydddd wants to merge2 commits intometa-pytorch:main
base:main
Choose a base branch
Loading
fromjoydddd:flex_att
Open

Conversation

@joydddd
Copy link

@joyddddjoydddd commentedJul 30, 2024
edited
Loading

Implement gpt-fast using flex_attention HOP.

replies on this PR:pytorch/pytorch#132157

@facebook-github-botfacebook-github-bot added the CLA SignedThis label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. labelJul 30, 2024
@joydddd
Copy link
Author

joydddd commentedJul 30, 2024
edited
Loading

python generate.py --compile --max_new_tokens 1024  --checkpoint_path checkpoints/$MODEL_REPO/model.pth --prompt "Hello, my name is"

sdpa_kernel:
Average tokens/sec: 137.30
Memory used: 13.91 GB
->
flex_attetion:
Average tokens/sec: 131.95
Memory used: 14.55 GB

@facebook-github-bot

Hi@joydddd!

Thank you for your pull request.

Werequire contributors to sign ourContributor License Agreement, and yours needs attention.

You currently have a record in our system, but the CLA is no longer valid, and will need to beresubmitted.

Process

In order for us to review and merge your suggested changes, please sign athttps://code.facebook.com/cla.If you are contributing on behalf of someone else (eg your employer), the individual CLA may not be sufficient and your employer may need to sign the corporate CLA.

Once the CLA is signed, our tooling will perform checks and validations. Afterwards, thepull request will be tagged withCLA signed. The tagging process may take up to 1 hour after signing. Please give it that time before contacting us about it.

If you have received this in error or have any questions, please contact us atcla@meta.com. Thanks!

Sign up for freeto join this conversation on GitHub. Already have an account?Sign in to comment

Reviewers

No reviews

Assignees

No one assigned

Labels

CLA SignedThis label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@joydddd@facebook-github-bot

[8]ページ先頭

©2009-2025 Movatter.jp