triton-ascend-case-matmul-large-k
OfficialSpeed up large-K matmul with Split-K parallelism.
Authormindspore-ai
Version1.0.0
Installs0
System Documentation
What problem does it solve?
Optimizes large-K matrix multiplication by splitting the K dimension to keep all cores busy, reducing global synchronization and improving throughput for cases where K >> M and N.
Core Features & Use Cases
- Split-K parallelization across grid dimensions to maximize core utilization and throughput.
- Atomic-add accumulation of partial results to avoid costly barriers.
- Workspace-based reduction option to minimize kernel-wide synchronization.
- Target scenario: M and N around 256 with K around 131072, where traditional matmul stalls.
Quick Start
Run the matmul_large_k example with M=256, K=131072, N=256 to observe Split-K parallelism gains.
Dependency Matrix
Required Modules
None requiredComponents
Standard package💻 Claude Code Installation
Recommended: Let Claude install automatically. Simply copy and paste the text below to Claude Code.
Please help me install this Skill: Name: triton-ascend-case-matmul-large-k Download link: https://github.com/mindspore-ai/akg/archive/main.zip#triton-ascend-case-matmul-large-k Please download this .zip file, extract it, and install it in the .claude/skills/ directory.
Agent Skills Search Helper
Install a tiny helper to your Agent, search and equip skill from 471,000+ vetted skills library on demand.