欢迎来到 Triton 的文档!

Triton 是一个用于并行编程的语言和编译器。它旨在提供一个基于 Python 的编程环境,用于高效地编写自定义的深度神经网络计算内核,这些内核能够在现代 GPU 硬件上以最大吞吐量运行。

入门

  • 请根据您选择的平台,遵循安装说明进行操作。

  • 查阅教程,了解如何编写您的第一个 Triton 程序。

Python API

Triton MLIR 方言和操作

深入了解

查阅以下文档,进一步了解 Triton 以及它与其他 DNN 领域 DSL 的对比