hip-sys

Rust 对 HIP 的绑定

依赖项

• ROCm 平台，即兼容的 AMD GPU
  • hipblas 可以通过 blas 功能访问
  • 使用 HIP_PATH 指定 hip 的路径。如果未提供，则假定 /opt/rocm/hip
  • 使用 HIP_BLAS_PATH 指定 hipblas 的路径。如果未提供，则假定 /opt/rocm/hipblas
  • 可以使用 bindgen 功能生成本地绑定
  • 在没有 AMD GPU 的情况下也可以编译，但设备函数将失败
  • 目前尚不支持 CUDA 平台（参见 cuda-sys https://github.com/rust-cuda/cuda-sys）

安装 ROCm

• Ubuntu：请参阅 https://rocmdocs.amd.com/en/latest/deploy/linux/quick_start.html
  • 与 hcc 后端兼容 sudo apt install hip-hcc（来自 ROCm 软件包注册表）
  • 您可能还需要安装 rocsolver sudo apt install rocsolver
• Arch：请参阅 https://github.com/arch4edu/arch4edu
  • sudopacman -Srocm-hip-runtime

如果您有任何问题，请发布一个问题。

测试

使用以下命令运行测试

cargo test

文档

使用以下命令打开文档

cargo doc --open

致谢

此仓库是从这里创建的 here；已获得发布此工作的许可（请参阅问题 #1）。