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

PyTorch集成FlyDSL加速AMD MI350 GPU

1 篇报道1 个报道来源2 小时前更新

先了解这件事

AI 综述

AMD和PyTorch团队集成了FlyDSL,使PyTorch用户能在AMD MI350系列GPU上通过现有算子API加速特定计算操作,无需调用自定义内核。 该支持目前针对采用gfx950架构的AMD MI350系列GPU,用户需使用最新的ROCm PyTorch nightly版本,并通过pip安装指定版本的FlyDSL库。

AI 根据报道生成 · 1 小时前更新

报道时间线

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

10月22日
  1. PyTorch 博客
    PyTorch 通过 FlyDSL 加速 AMD GPU 计算

    PyTorch 用户可在 AMD MI350 系列 GPU 上通过 FlyDSL 使用 dense 和 grouped GEMM、fused multi-head attention 等操作,无需调用自定义内核。

本事件热度走势

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