Skip to content

feat(tilelang): add TileLang backend support#215

Open
zihaomu wants to merge 2 commits into
AMD-AGI:mainfrom
zihaomu:pr2-clean
Open

feat(tilelang): add TileLang backend support#215
zihaomu wants to merge 2 commits into
AMD-AGI:mainfrom
zihaomu:pr2-clean

Conversation

@zihaomu
Copy link
Copy Markdown

@zihaomu zihaomu commented May 13, 2026

Summary

Adds TileLang as a supported kernel backend in GEAK.

With this PR, GEAK can recognize TileLang kernels, route them through the heterogeneous optimization flow, generate TileLang-aware optimization guidance, and handle TileLang test/harness discovery more reliably.

Known limitation / follow-up
TileLang-specific profiling support is not fully included in this PR yet.

Example

For a simple TileLang kernel like this:

import tilelang
import tilelang.language as T

@tilelang.jit
def add(N: int, block: int = 256):
    @T.prim_func
    def add_kernel(
        A: T.Tensor((N,), "float32"),
        B: T.Tensor((N,), "float32"),
        C: T.Tensor((N,), "float32"),
    ):
        with T.Kernel(T.ceildiv(N, block), threads=block) as bx:
            for i in T.Parallel(block):
                idx = bx * block + i
                C[idx] = A[idx] + B[idx]

    return add_kernel

Co-authored-by: Zheng Li zli10002@amd.com

@zihaomu zihaomu changed the title feat(tilelang): add TileLang backend support feat(tilelang): add TileLang backend support (WIP) May 13, 2026
zihaomu and others added 2 commits May 13, 2026 10:00
Co-authored-by: zli10002 <zli10002@amd.com>
Co-authored-by: zli10002 <zli10002@amd.com>
@zihaomu zihaomu changed the title feat(tilelang): add TileLang backend support (WIP) feat(tilelang): add TileLang backend support May 14, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant