tilelang
CommunityPowerful GPU kernel development with automated high-performance code generation
Authorjstzwj
Version1.0.0
Installs0
System Documentation
What problem does it solve?
TileLang simplifies the development of high-performance GPU kernels by providing a framework that automatically dispatches to hardware-optimized instructions, reducing complexity for engineers.
Core Features & Use Cases
- Kernel Automation: Generates efficient code for CUDA, HIP, Metal, and CPU architectures with minimal manual tuning.
- Deep API Access: Offers primitives for memory management, data movement, tensor core operations, and control flow.
- Use Case: Develops optimized GEMM kernels and advanced tensor operations for AI model training and inference, streamlining workflows and maximizing hardware utilization.
Quick Start
Write a TileLang kernel to perform matrix multiplication, compile, and execute it directly within your Python environment.
Dependency Matrix
Required Modules
tvmtorchnumpy
Components
scriptsreferencesassets
💻 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: tilelang Download link: https://github.com/jstzwj/ai-infra-plugins/archive/main.zip#tilelang 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.