Skip to content

[CK_TILE] Support for CompV4 pipeline in Stream-K GEMM#5445

Open
arai713 wants to merge 5 commits intodevelopfrom
users/arai/ck/streamk_compv4
Open

[CK_TILE] Support for CompV4 pipeline in Stream-K GEMM#5445
arai713 wants to merge 5 commits intodevelopfrom
users/arai/ck/streamk_compv4

Conversation

@arai713
Copy link
Contributor

@arai713 arai713 commented Mar 13, 2026

Motivation

This PR is extending the pipeline support for Stream-K GEMM by adding the CompV4 pipeline. Additional pipelines will be added in subsequent PRs.

Technical Details

  • Enable the CompV4 pipeline by adding an option to set DoubleSMemBuffer to true if the CompV4 pipeline has been selected as it requires double buffered shared memory
  • Addition of CompV4 pipeline into the extended tests: kernel instances mirror the existing CompV3/Mem configurations (same layout permutations, data types, and tile sizes) with the pipeline type set to CompV4.
  • Addition of CompV4 pipeline into smoke tests (generated using Tile Engine)

Test Plan

These were tested using the existing smoke and extended tests.

Test Result

All tests passed

Submission Checklist

@arai713 arai713 force-pushed the users/arai/ck/streamk_compv4 branch from 22ade29 to 320ef34 Compare March 16, 2026 17:38
Copy link
Contributor

@cgmillette cgmillette left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM. Just need to address the CI failures.

@cgmillette cgmillette enabled auto-merge (squash) March 18, 2026 15:40
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