跳到正文
原文
OpenAI News·· 2017-12-06精选AI 评分62

OpenAI 发布块稀疏 GPU kernel

Block-sparse GPU kernels

AI 导读

OpenAI 发布针对块稀疏权重神经网络的高度优化 GPU kernel,按所选稀疏度可比 cuBLAS 或 cuSPARSE 快数个数量级。OpenAI 表示已用这些 kernel 在文本情感分析和文本、图像生成建模上取得当时的最优结果。

推荐理由

OpenAI 公开面向块稀疏权重网络的 GPU kernel,并给出相对 cuBLAS 与 cuSPARSE 的加速幅度和落地任务。

来源:OpenAI News · openai.com