Update fence_proxy API to string-based arguments in cutlass_api GEMM … - #3364
Closed
nWEIdia wants to merge 1 commit into
Closed
Update fence_proxy API to string-based arguments in cutlass_api GEMM …#3364nWEIdia wants to merge 1 commit into
nWEIdia wants to merge 1 commit into
Conversation
…impls
Replace deprecated cute.arch.ProxyKind/SharedSpace enum usage with
the new string-based fence_proxy API: fence_proxy("async.shared", space="cta").
Co-authored-by: Wei Wang <nWEIdia@users.noreply.github.com>
Author
|
cc @hwu36 could you please help review and/or redirect? Thanks! |
sahnimanas
approved these changes
Jul 7, 2026
sahnimanas
left a comment
sahnimanas
left a comment
There was a problem hiding this comment.
LGTM as a stop-gap.
Recording for posterity here that cutlass_api is a side-branch for preview, and official support will come as CUTLASS Operator API.
Author
|
Closing, since pytorch/pytorch#189772 landed. |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
3 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Update fence_proxy API to string-based arguments in cutlass_api GEMM impls
Replace deprecated cute.arch.ProxyKind/SharedSpace enum usage with the new string-based fence_proxy API: fence_proxy("async.shared", space="cta").
Credit goes to Cursor Coding agent.
Background
Issue: pytorch/pytorch#188477
Related PyTorch PR: pytorch/pytorch#188633
Note: this is a temporary solution to unblock PyTorch by fixing the failed tests. The longer term is to move away from "cutlass_api" branch usage and migrate PyTorch with it to e.g. cutlass operators.