Python DSL 添加 LSP 总共分几步

主题演讲
上海
16:30 - 17:10
分会场 E(综合楼 108 报告厅)
  • 王文洋 AI Infer 工程师,GitHub 街溜子

    wangwenyang

摘要

从 TileLang 出发,看看为一门"长得像 Python”却写起来不像 Python 的算子 DSL 增加 LSP 支持,总共要走几步。

详情

随着大模型训练、推理需求的膨胀,一批为了简化算子编写的 DSL 被发明出来,例如 Triton、TileLang、CuTeDSL。它们长得像 Python、用起来像 Python,但是写起来一点也不像 Python。为了提高编写和阅读体验,增加 LSP 支持,提供如类型推导、化简、报错等功能,便是一个顺理成章的想法。

让我们从 TileLang 开始,看看为它增加 LSP 的支持总共分几步。