Skip to content

Poc gpu tileir conversion main sync同步 Triton 3.6 主线并完善 TLE GPU CommonIR 到 TTGIR 转换 - #1086

Open
macqueen09 wants to merge 10 commits into
flagos-ai:mainfrom
KernelLLM:poc-gpu-tileir-conversion-main-sync
Open

Poc gpu tileir conversion main sync同步 Triton 3.6 主线并完善 TLE GPU CommonIR 到 TTGIR 转换#1086
macqueen09 wants to merge 10 commits into
flagos-ai:mainfrom
KernelLLM:poc-gpu-tileir-conversion-main-sync

Conversation

@macqueen09

Copy link
Copy Markdown
  • 基线 Commit:3f863acd700bda89514cf3dce6c57020d593d8f3
  • 当前分支:poc-gpu-tileir-conversion-main-sync
  • 当前 Commit:dda7eefda2ffacec39a51821a47f1d6fb148bdf5

在完成主线同步后,重新叠加了 TLE GPU CommonIR 相关修改,主要包括:

  1. tle.gpu.alloc/copy/to_tensor/store_tensor/local_ptr 等原语先生成 TileIR;
  2. 增加显式的 TileIR → TTGIR conversion;
  3. !tile.buf<#shared> 转换为 !ttg.memdescttg.local_alloc/load/store
  4. 支持 tile.local_ptr/get_memdesc/subview 等中间表示的转换;
  5. 使用 FLAGTREE_COMMON_IR 编译宏控制 CommonIR 路径,关闭时保留原有 TLE GPU 路径;
  6. 补充矩阵乘法、FlashAttention 和 TLE GPU 基础原语测试。

FLIR 配套版本

本 PR 需要配合以下 FLIR 修改:

  • FLIR PR:kateyijian/flir#14
  • 分支:poc-gpu-tileir-conversion
  • Commit:3f3ac6fccc91cd8434c8d055da28cd1ca94974bf
  • 目标分支:main

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.

3 participants