ROCm的英文全称Radeon Open Compute platform,它是AMD在去年12月推出的一款开源GPU运算平台,目前已经发展到了1.3版本。MIOpen则是AMD为此开发的软件库,其作用是将程序设计语言和ROCm平台连接,以充分利用GCN架构。
本次发布的版本包括以下内容:
- 深度卷积解算器针对前向和后向传播进行了优化
- 包括Winograd和FFT转换在内的优化卷积
- 优化GEMM深入学习
- Pooling、Softmax、Activations、梯度算法的批量标准化和LR规范化
- MIOpen将数据描述为4-D张量 - Tensors 4D NCHW格式
- 支持OpenCL和HIP的框架API
- MIOpen驱动程序可以测试MIOpen中任何特定图层的向前/向后网络
- 二进制包增加了对Ubuntu 16.04和Fedora 24的支持
- 源代码位于https://github.com/ROCmSoftwarePlatform/MIOpen
- 参考文档
ROCm 1.6具有MIOpen的预构建软件包
安装ROCm MIOpen实现(这是建立在已经安装了“ROCm”和“ROCm-opencldev”包的前提下):
仅支持OpenCL开发
sudo apt-get install miopengemm miopen-opencl
仅支持HIP开发
sudo apt-get install miopengemm miopen-hip
或者你可以从源代码中构建