基于 Python 的矩阵运算加速器

Xilinx GEMX(通用矩阵运算)库可提供一套高性能引擎,用于加速严重依赖矩阵运算的应用。该库附带了一组能够支持软件的 Python API,特别是 Python 开发人员可以轻松利用这些引擎的性能优势。

主要特性与优势

  • 高性能密集和稀疏矩阵运算加速器
  • 通过指令控制的引擎减少数据移动开销
  • 支持行主格式密集矩阵运算
  • 支持 COO 格式稀疏矩阵运算
  • 支持全面连接的网络操作
  • 易于使用且高效的Python API
  • 提供加速器卡支持

    推荐阅读