跳到正文
热点事件持续更新

PyTorch 推出 CUDA Tile 工具

1 篇报道1 个报道来源51 分钟前更新

先了解这件事

AI 综述

PyTorch 官方推出 CUDA Tile,这是一套面向 Python 的领域专用语言(DSL)和以 tile 为中心的内核原语,可将原本分开计算多项的直接实现融合为单个 tiled kernel,并把矩阵保留在寄存器中。官方称,该工具能自动针对 Tensor Core 等高级 GPU 硬件特性优化内核,同时提供示例、文档和样例代码。

AI 根据报道生成 · 22 分钟前更新

报道时间线

沿着报道,了解事件的不同侧面。

11月10日
  1. AGI Hunt
    PyTorch 推出 CUDA Tile Python DSL 与 tile 级内核原语

    PyTorch 官方介绍 CUDA Tile,一套基于 Python 的 DSL 和 tile 级原语,可自动面向 Tensor Core 等高级 GPU 硬件特性优化内核。该方案能把 PyTorch 中分开计算多项的直接实现融合成单个 tiled kernel,并将矩阵保留在寄存器中,官方同时提供了示例、文档和样例代码。

本事件热度走势

还没有足够的连续观测数据,暂不绘制趋势。