Triton 简介与并行编程基础
并行编程基础知识
Triton介绍
基于Triton 的算子开发
Triton 语法
triton 和 triton.language 模组
常见计算内核
基于Triton 的大模型推理
结合pytorch 使用triton 编写的计算内核 进行大模型推理
九齿与面向张量的元编程
面向张量的元编程
九齿并行编程模型和 领域特定语言
基于九齿的算子开发
实现常见计算内核
适用环境
-
官方支持的操作系统
-
Linux
-
官方支持的硬件
-
NVIDIA GPUs (Compute Capability 8.0+)
-
AMD GPUs (ROCm 5.2+)
-
CPUs (开发中)
-
第三方支持的硬件:
-
摩尔线程的 GPU
-
天数智芯的 GPU
-
沐曦的 GPU
-
……


评论(0)
暂无评论