NVIDIA

cutile-python

NVIDIA

cuTile is a programming model for writing parallel kernels for NVIDIA GPUs

AI 简介

cuTile Python 是一个面向 NVIDIA GPU 的并行核编程模型,用于编写基于分块(tiling)的高性能 CUDA 内核。它提供 Python 原生语法接口(如 @ct.kernel 装饰器),通过 Tile IR 中间表示自动生成优化的 GPU 代码,支持显式内存加载/存储与分块调度,并依赖 tileiras 编译器后端。核心特点包括声明式分块操作、与 CuPy 紧密集成、适配 Ampere/Ada/Blackwell 架构 GPU。适用于需要细粒度控制 GPU 内存访问和计算调度的场景,如定制化算子开发、高性能数值计算及 AI 推理加速等。

Python
Other
2.1k
Stars
142
Forks
21
Watchers
19
Issues

Star 增长

今日0
近 7 天0
近 30 天0
综合评分58.47
默认分支main