5.1。概述

原文： http://numba.pydata.org/numba-doc/latest/roc/overview.html

Numba 支持 AMD ROC GPU 编程，直接将 HSA 代码的受限子集编译为 HSA 内核和遵循 HSA 执行模型的设备功能。用 Numba 编写的内核似乎可以直接访问 NumPy 数组。

5.1.1。术语

这里列出了 HSA 编程主题中的几个重要术语：

  • 内核 ：由主机启动并在设备上执行的 GPU 功能
  • 设备功能 ：在设备上执行的 GPU 功能，只能从设备调用（即从内核或其他设备功能）

5.1.2。要求

本文件描述了使用 ROC 的要求。基本上需要 AMD dGPU（斐济，Polaris 和 Vega 系列）以及支持 PCIe Gen3 和 PCIe Atomics 的 CPU（AMD Ryzen 和 EPYC 以及 Intel CPU＆gt; = Haswell），完整的详细信息在链接文档中。此外，还需要 Linux 操作系统，支持和测试的操作系统也列在链接文档中。

5.1.3。安装

请按照此文档获取安装说明，以启用系统的 ROC 支持。请务必使用系统的 Linux 发行版的二进制包来简化该过程。此时应通过运行来测试安装：

  1. $ /opt/rocm/bin/rocminfo

其输出应列出至少两个 HSA 代理，其中至少一个应为 CPU，其中至少一个应为 dGPU。

假设安装工作正常，NOC 的 ROC 支持由roctools包提供，可以通过conda和 Numba 从 Numba 通道安装如下（创建一个名为numba_roc的 env）：

  1. $ conda create -n numba_roc -c numba numba roctools

激活 env，然后运行 Numba 诊断工具应确认 Numba 在启用 ROC 支持的情况下运行，例如：

  1. $ source activate numba_roc
  2. $ numba -s

numba -s的输出应包含类似于的部分：

  1. __ROC Information__
  2. ROC available                       : True
  3. Available Toolchains                : librocmlite library, ROC command line tools
  5. Found 2 HSA Agents:
  6. Agent id  : 0
  7.     vendor: CPU
  8.     name: Intel(R) Core(TM) i7-4790 CPU @ 3.60GHz
  9.     type: CPU
  11. Agent id  : 1
  12.     vendor: AMD
  13.     name: gfx803
  14.     type: GPU
  16. Found 1 discrete GPU(s)             : gfx803

确认 ROC 可用，列出可用的工具链并显示 HSA 代理和 dGPU 计数。