aiter-ck-gemm-tune

Automate tuning of AITER CK GEMM and fused MoE kernels on AMD GPUs.

1|Updated Feb 7, 2026
One-click install
npx skills add https://github.com/alexsun07/agent_skills --skill aiter-ck-gemm-tune
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: aiter-ck-gemm-tune
Source: https://github.com/alexsun07/agent_skills/tree/main/skills/aiter-ck-gemm-tune
Command: npx skills add https://github.com/alexsun07/agent_skills --skill aiter-ck-gemm-tune

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires aiter, rocminfo, and includes scripts (resource) and references (resource) components.

What problem does it solve?

This Skill automates the tuning of AITER's Composable Kernel (CK) GEMM and fused MoE kernels, improving performance for specific model shapes on AMD GPUs.

Core Features & Use Cases

  • Kernel Tuning: Automatically discover, benchmark, and tune GEMM and MoE kernel configurations for optimal performance.
  • Shape Discovery: Extracts model shapes from inference logs for tuning.
  • Benchmarking: Compares performance before and after tuning.
  • Use Case: Ideal for optimizing the performance of LLM inference on ROCm/AMD GPUs.

Quick Start

Use the aiter-ck-gemm-tune skill to tune the GEMM and MoE kernels for your model shapes.

Frequently Asked Questions about aiter-ck-gemm-tune

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I tune AITER CK GEMM kernels for specific model shapes on AMD GPUs?

To tune AITER CK GEMM kernels, you can automate the discovery and benchmarking of kernel configurations to optimize performance for specific model shapes on AMD GPUs. This process extracts shapes from inference logs to find the optimal setup.

What is the best way to optimize fused MoE kernel performance for LLM inference on ROCm?

Optimizing fused MoE kernel performance on ROCm involves automating the tuning process to benchmark and compare configurations before and after adjustments. This ensures optimal LLM inference efficiency on AMD GPUs.

Do I need AITER installed to automate Composable Kernel tuning for GEMM operations?

Yes, you need the AITER package installed along with rocminfo. These dependencies are required to automate the kernel configuration and benchmarking process for Composable Kernel GEMM operations.

Can I extract model shapes from inference logs for AMD GPU kernel tuning?

Yes, you can extract model shapes directly from inference logs. The tuning process uses these extracted shapes to discover, benchmark, and apply the best Composable Kernel GEMM and fused MoE configurations.

Does this kernel tuning process support performance benchmarking before and after optimization?

Yes, the kernel tuning process includes benchmarking functionality that compares performance metrics before and after tuning. This allows you to verify the optimization gains for your specific GEMM and MoE kernels.