Skip to content

fp8 rowwise regular gemm tuning for llm new shapes #3654

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

Closed
wants to merge 1 commit into from

Conversation

mxz297
Copy link
Contributor

@mxz297 mxz297 commented Feb 3, 2025

Summary:
X-link: https://github.com/facebookresearch/FBGEMM/pull/730

  1. Add more FP8 rowwise instances
  2. Extend the template to allow specifying AK1 and BK1
  3. Add tuning for new shapes

Differential Revision: D69070084

Summary:
X-link: facebookresearch/FBGEMM#730

1. Add more FP8 rowwise instances
2. Extend the template to allow specifying AK1 and BK1
3. Add tuning for new shapes

Differential Revision: D69070084
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D69070084

Copy link

netlify bot commented Feb 3, 2025

Deploy Preview for pytorch-fbgemm-docs ready!

Name Link
🔨 Latest commit b352fb2
🔍 Latest deploy log https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/67a12bc7b11d8000083e3886
😎 Deploy Preview https://deploy-preview-3654--pytorch-fbgemm-docs.netlify.app
📱 Preview on mobile
Toggle QR Code...

QR Code

Use your smartphone camera to open QR code link.

To edit notification comments on pull requests, go to your Netlify site configuration.

@facebook-github-bot
Copy link
Contributor

This pull request has been merged in bdcce9c.

avbokovoy pushed a commit to ROCm/FBGEMM that referenced this pull request Feb 14, 2025
Summary:
Pull Request resolved: pytorch#3654

X-link: https://github.com/facebookresearch/FBGEMM/pull/730

1. Add more FP8 rowwise instances
2. Extend the template to allow specifying AK1 and BK1
3. Add tuning for new shapes

Reviewed By: jianyuh

Differential Revision: D69070084

fbshipit-source-id: 7f841bd0855743d5389d7afa0f135619ca0cd61a
q10 pushed a commit to q10/FBGEMM that referenced this pull request Apr 10, 2025
Summary:
X-link: pytorch#3654

Pull Request resolved: facebookresearch/FBGEMM#730

1. Add more FP8 rowwise instances
2. Extend the template to allow specifying AK1 and BK1
3. Add tuning for new shapes

Reviewed By: jianyuh

Differential Revision: D69070084

fbshipit-source-id: 7f841bd0855743d5389d7afa0f135619ca0cd61a
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants