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