# 5.1。概述
> 原文: [http://numba.pydata.org/numba-doc/latest/roc/overview.html](http://numba.pydata.org/numba-doc/latest/roc/overview.html)
Numba 支持 [AMD ROC GPU](https://rocm.github.io/) 编程,直接将 HSA 代码的受限子集编译为 HSA 内核和遵循 HSA 执行模型的设备功能。用 Numba 编写的内核似乎可以直接访问 NumPy 数组。
## 5.1.1。术语
这里列出了 HSA 编程主题中的几个重要术语:
* _ 内核 _:由主机启动并在设备上执行的 GPU 功能
* _ 设备功能 _:在设备上执行的 GPU 功能,只能从设备调用(即从内核或其他设备功能)
## 5.1.2。要求
[本文件](https://github.com/RadeonOpenCompute/ROCm#are-you-ready-to-rock)描述了使用 ROC 的要求。基本上需要 AMD dGPU(斐济,Polaris 和 Vega 系列)以及支持 PCIe Gen3 和 PCIe Atomics 的 CPU(AMD Ryzen 和 EPYC 以及 Intel CPU> = Haswell),完整的详细信息在链接文档中。此外,还需要 Linux 操作系统,支持和测试的操作系统也列在链接文档中。
## 5.1.3。安装
请按照[此文档](https://github.com/RadeonOpenCompute/ROCm#installing-from-amd-rocm-repositories)获取安装说明,以启用系统的 ROC 支持。请务必使用系统的 Linux 发行版的二进制包来简化该过程。此时应通过运行来测试安装:
```py
$ /opt/rocm/bin/rocminfo
```
其输出应列出至少两个 HSA 代理,其中至少一个应为 CPU,其中至少一个应为 dGPU。
假设安装工作正常,NOC 的 ROC 支持由`roctools`包提供,可以通过`conda`和 Numba 从 Numba 通道安装如下(创建一个名为`numba_roc`的 env):
```py
$ conda create -n numba_roc -c numba numba roctools
```
激活 env,然后运行 Numba 诊断工具应确认 Numba 在启用 ROC 支持的情况下运行,例如:
```py
$ source activate numba_roc
$ numba -s
```
`numba -s`的输出应包含类似于的部分:
```py
__ROC Information__
ROC available : True
Available Toolchains : librocmlite library, ROC command line tools
Found 2 HSA Agents:
Agent id : 0
vendor: CPU
name: Intel(R) Core(TM) i7-4790 CPU @ 3.60GHz
type: CPU
Agent id : 1
vendor: AMD
name: gfx803
type: GPU
Found 1 discrete GPU(s) : gfx803
```
确认 ROC 可用,列出可用的工具链并显示 HSA 代理和 dGPU 计数。
- 1. 用户手册
- 1.1。 Numba 的约 5 分钟指南
- 1.2。概述
- 1.3。安装
- 1.4。使用@jit 编译 Python 代码
- 1.5。使用@generated_jit 进行灵活的专业化
- 1.6。创建 Numpy 通用函数
- 1.7。用@jitclass 编译 python 类
- 1.8。使用@cfunc 创建 C 回调
- 1.9。提前编译代码
- 1.10。使用@jit 自动并行化
- 1.11。使用@stencil装饰器
- 1.12。从 JIT 代码 中回调到 Python 解释器
- 1.13。性能提示
- 1.14。线程层
- 1.15。故障排除和提示
- 1.16。常见问题
- 1.17。示例
- 1.18。会谈和教程
- 2. 参考手册
- 2.1。类型和签名
- 2.2。即时编译
- 2.3。提前编译
- 2.4。公用事业
- 2.5。环境变量
- 2.6。支持的 Python 功能
- 2.7。支持的 NumPy 功能
- 2.8。与 Python 语义的偏差
- 2.9。浮点陷阱
- 2.10。 Python 2.7 寿命终止计划
- 3. 用于 CUDA GPU 的 Numba
- 3.1。概述
- 3.2。编写 CUDA 内核
- 3.3。内存管理
- 3.4。编写设备功能
- 3.5。 CUDA Python 中支持的 Python 功能
- 3.6。支持的原子操作
- 3.7。随机数生成
- 3.8。设备管理
- 3.10。示例
- 3.11。使用 CUDA 模拟器 调试 CUDA Python
- 3.12。 GPU 减少
- 3.13。 CUDA Ufuncs 和广义 Ufuncs
- 3.14。共享 CUDA 内存
- 3.15。 CUDA 阵列接口
- 3.16。 CUDA 常见问题
- 4. CUDA Python 参考
- 4.1。 CUDA 主机 API
- 4.2。 CUDA 内核 API
- 4.3。内存管理
- 5. 用于 AMD ROC GPU 的 Numba
- 5.1。概述
- 5.2。编写 HSA 内核
- 5.3。内存管理
- 5.4。编写设备功能
- 5.5。支持的原子操作
- 5.6。代理商
- 5.7。 ROC Ufuncs 和广义 Ufuncs
- 5.8。示例
- 6. 扩展 Numba
- 6.1。高级扩展 API
- 6.2。低级扩展 API
- 6.3。示例:间隔类型
- 7. 开发者手册
- 7.1。贡献给 Numba
- 7.2。 Numba 建筑
- 7.3。多态调度
- 7.4。关于发电机的注意事项
- 7.5。关于 Numba Runtime 的注意事项
- 7.6。使用 Numba Rewrite Pass 获得乐趣和优化
- 7.7。实时变量分析
- 7.8。上市
- 7.9。模板注释
- 7.10。关于自定义管道的注意事项
- 7.11。环境对象
- 7.12。哈希 的注意事项
- 7.13。 Numba 项目路线图
- 8. Numba 增强建议
- 9. 术语表