Skip to content

Optimize wgrad CUTLASS grouped gemm#4891

Closed
jiawenliu64 wants to merge 3 commits into
mainfrom
export-D82700455
Closed

Optimize wgrad CUTLASS grouped gemm#4891
jiawenliu64 wants to merge 3 commits into
mainfrom
export-D82700455

Conversation

@jiawenliu64

Copy link
Copy Markdown

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

  • Make wgrad CUTLASS grouped gemm return float32 output when wgrad is provided, respecting e2e
  • Optimize general heuristic
  • Make tests cover wgrad accum with float32 output

Differential Revision: D82700455

@netlify

netlify Bot commented Sep 18, 2025

Copy link
Copy Markdown

Deploy Preview for pytorch-fbgemm-docs ready!

Name Link
🔨 Latest commit 9cd2d85
🔍 Latest deploy log https://app.netlify.com/projects/pytorch-fbgemm-docs/deploys/68d2ce446d3d8c00093f4e79
😎 Deploy Preview https://deploy-preview-4891--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 project configuration.

@meta-cla meta-cla Bot added the cla signed label Sep 18, 2025
@facebook-github-bot

Copy link
Copy Markdown
Contributor

@jiawenliu64 has exported this pull request. If you are a Meta employee, you can view the originating diff in D82700455.

@facebook-github-bot

Copy link
Copy Markdown
Contributor

@jiawenliu64 has exported this pull request. If you are a Meta employee, you can view the originating diff in D82700455.

jiawenliu64 pushed a commit that referenced this pull request Sep 23, 2025
Summary:
Pull Request resolved: #4891

X-link: facebookresearch/FBGEMM#1916

- Make wgrad CUTLASS grouped gemm return float32 output when wgrad is provided, respecting e2e
- Optimize general heuristic
- Make tests cover wgrad accum with float32 output

Reviewed By: q10

Differential Revision: D82700455
@facebook-github-bot

Copy link
Copy Markdown
Contributor

@jiawenliu64 has exported this pull request. If you are a Meta employee, you can view the originating diff in D82700455.

jiawenliu64 pushed a commit that referenced this pull request Sep 23, 2025
Summary:
Pull Request resolved: #4891

X-link: facebookresearch/FBGEMM#1916

- Make wgrad CUTLASS grouped gemm return float32 output when wgrad is provided, respecting e2e
- Optimize general heuristic
- Make tests cover wgrad accum with float32 output

Reviewed By: q10

Differential Revision: D82700455
@facebook-github-bot

Copy link
Copy Markdown
Contributor

@jiawenliu64 has exported this pull request. If you are a Meta employee, you can view the originating diff in D82700455.

jiawenliu64 pushed a commit that referenced this pull request Sep 23, 2025
Summary:
Pull Request resolved: #4891

X-link: facebookresearch/FBGEMM#1916

- Make wgrad CUTLASS grouped gemm return float32 output when wgrad is provided, respecting e2e
- Optimize general heuristic
- Make tests cover wgrad accum with float32 output

Reviewed By: q10

Differential Revision: D82700455
jiawenl added 2 commits September 23, 2025 09:05
@facebook-github-bot

Copy link
Copy Markdown
Contributor

@jiawenliu64 has exported this pull request. If you are a Meta employee, you can view the originating diff in D82700455.

jiawenliu64 pushed a commit that referenced this pull request Sep 23, 2025
Summary:
Pull Request resolved: #4891

X-link: facebookresearch/FBGEMM#1916

- Make wgrad CUTLASS grouped gemm return float32 output when wgrad is provided, respecting e2e
- Optimize general heuristic
- Make tests cover wgrad accum with float32 output

Reviewed By: q10

Differential Revision: D82700455
@facebook-github-bot

Copy link
Copy Markdown
Contributor

@jiawenliu64 has exported this pull request. If you are a Meta employee, you can view the originating diff in D82700455.

jiawenliu64 pushed a commit that referenced this pull request Sep 23, 2025
Summary:
Pull Request resolved: #4891

X-link: facebookresearch/FBGEMM#1916

- Make wgrad CUTLASS grouped gemm return float32 output when wgrad is provided, respecting e2e
- Optimize general heuristic
- Make tests cover wgrad accum with float32 output

Reviewed By: q10

Differential Revision: D82700455
@facebook-github-bot

Copy link
Copy Markdown
Contributor

@jiawenliu64 has exported this pull request. If you are a Meta employee, you can view the originating diff in D82700455.

jiawenliu64 pushed a commit that referenced this pull request Sep 23, 2025
Summary:
Pull Request resolved: #4891

X-link: facebookresearch/FBGEMM#1916

- Make wgrad CUTLASS grouped gemm return float32 output when wgrad is provided, respecting e2e
- Optimize general heuristic
- Make tests cover wgrad accum with float32 output

Reviewed By: q10

Differential Revision: D82700455
Summary:
Pull Request resolved: #4891

X-link: facebookresearch/FBGEMM#1916

- Make wgrad CUTLASS grouped gemm return float32 output when wgrad is provided, respecting e2e
- Optimize general heuristic
- Make tests cover wgrad accum with float32 output

Reviewed By: q10

Differential Revision: D82700455
@facebook-github-bot

Copy link
Copy Markdown
Contributor

@jiawenliu64 has exported this pull request. If you are a Meta employee, you can view the originating diff in D82700455.

@facebook-github-bot

Copy link
Copy Markdown
Contributor

This pull request has been merged in ddada9e.

@spcyppt spcyppt deleted the export-D82700455 branch March 3, 2026 03:09
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.

2 participants