Copyright (c) 2012-Present, Syoyo Fujita and many contributors.
所有只涉及 constexpr 值的 Python 逻辑在编译期完全展开和消除,使得循环展开、分支消除等优化非常自然。 Triton DSL 的核心原语 (program_id 、 arange 、 load 、 store 、 reduce 、 dot 、 broadcast 、 trans 、 reshape 、 cat)构成了一个精简但完备的张量计算原语集,每个原语的设计都与 GPU ...