Skip to content

Add the CUTLASS backend for cuda.coop - #11044

Draft
tpn wants to merge 1 commit into
codex/cuda-coop-01-corefrom
codex/cuda-coop-02-cutlass
Draft

Add the CUTLASS backend for cuda.coop#11044
tpn wants to merge 1 commit into
codex/cuda-coop-01-corefrom
codex/cuda-coop-02-cutlass

Conversation

@tpn

@tpn tpn commented Aug 27, 2026

Copy link
Copy Markdown
Contributor

Why this is needed

The portable cuda.coop layer defines cooperative operations, but it does not
yet make them usable from CUTLASS DSL kernels. This stack layer adds the
CUTLASS compiler substrate and implements the initial load/store, reduce, and
scan cohort behind the same API introduced in #11035.

import cuda.coop.cutlass as coop

items = coop.load(coop.this_block(), source, coop.ThreadData(2))
total = coop.sum(coop.this_block(), items)
coop.store(coop.this_block(), destination, items)

What this PR includes

  • CUTLASS DSL 4.8 activation and automatic backend registration
  • lowering and provider generation for load/store, reduce/sum, and scan
  • provider compilation, final linking, scratch storage, and CUB/CUDAX routing
  • semantic, header, toolkit, and compiler cache identity with integrity and
    fork-safe locking

PCH and AOT land in stack layers 6 and 7.

Stack

2 of 7, based on #11035. The next layer is the Numba-CUDA-MLIR backend.

Local validation

  • 552 focused CUTLASS source tests passed across unit, provider compilation,
    linking, and GPU runtime coverage
  • 101 affected import/boundary tests passed after the final stack cleanup
  • source-root and root-free final-link GPU lanes passed in the local CUTLASS
    DSL 4.8 environment
  • source imports, compileall, changed-file pre-commit, and
    git diff --check passed

Wheel and installed-package qualification, GitHub CI, and automated AI review
are deferred for now.

Introduce the CUTLASS compiler substrate and provider lifecycle. Add the
load/store, reduce, and scan walking cohort.

Harden semantic and header identity, toolkit selection, artifact
integrity, and fork-safe cache locking before automatic registration.

Signed-off-by: Trent Nelson <trent@trent.me>
@copy-pr-bot

copy-pr-bot Bot commented Aug 27, 2026

Copy link
Copy Markdown
Contributor

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

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

Labels

None yet

Projects

Status: In Progress

Development

Successfully merging this pull request may close these issues.

1 participant