Skip to content

more fp8 tuning for decode and not need to pad#3576

Closed
mxz297 wants to merge 1 commit into
pytorch:mainfrom
mxz297:export-D68233557
Closed

more fp8 tuning for decode and not need to pad#3576
mxz297 wants to merge 1 commit into
pytorch:mainfrom
mxz297:export-D68233557

Conversation

@mxz297

@mxz297 mxz297 commented Jan 15, 2025

Copy link
Copy Markdown
Contributor

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

This diff includes:

  1. Allow specifying split-K in the common tempalate
  2. Add a few more instances
  3. Update tuning for some decode shapes
  4. Change default GEMMSpecialization default to Default (no padding is needed when input is contiguous and not strided)

Differential Revision: D68233557

@facebook-github-bot

Copy link
Copy Markdown
Contributor

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

@netlify

netlify Bot commented Jan 15, 2025

Copy link
Copy Markdown

Deploy Preview for pytorch-fbgemm-docs ready!

Name Link
🔨 Latest commit 2da4852
🔍 Latest deploy log https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/67883de2f4319b00085d573a
😎 Deploy Preview https://deploy-preview-3576--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.

Summary:

X-link: facebookresearch/FBGEMM#661

This diff includes:

1. Allow specifying split-K in the common tempalate
2. Add a few more instances
3. Update tuning for some decode shapes
4. Change default GEMMSpecialization default to Default (no padding is needed when input is contiguous and not strided)

Differential Revision: D68233557
mxz297 added a commit to mxz297/FBGEMM that referenced this pull request Jan 15, 2025
Summary:

X-link: facebookresearch/FBGEMM#661

This diff includes:

1. Allow specifying split-K in the common tempalate
2. Add a few more instances
3. Update tuning for some decode shapes
4. Change default GEMMSpecialization default to Default (no padding is needed when input is contiguous and not strided)

Differential Revision: D68233557
@mxz297 mxz297 force-pushed the export-D68233557 branch 2 times, most recently from 663ee1d to 2da4852 Compare January 15, 2025 22:59
@facebook-github-bot

Copy link
Copy Markdown
Contributor

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

1 similar comment
@facebook-github-bot

Copy link
Copy Markdown
Contributor

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

@facebook-github-bot

Copy link
Copy Markdown
Contributor

This pull request has been merged in 5c76d93.

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