Testing GPU compute on AMD APU, NixOS

Help
4

You shouldn’t be needing these at least for ROCm; I don’t have any ROCm-capable hardware around, but I think you just need the kernel module

:+1:

Anyone have a suggestion for what test program I should aim for,

Here’s a very modest one to start with, it’s what we use to sanity-check CUDA, and it’s expected to work on amd using zluda as well: https://github.com/NixOS/nixpkgs/pull/288644#pullrequestreview-1999383217 (looking at Hydra - Build 266084426 of job nixpkgs:trunk:zluda.x86_64-linux now, it’s been suffering form the timeouts in rocmPackages too)

Attempting nix-shell -p python311Packages.torchWithRocm results in a build error complaining about hipblaslt

Yes, this has been waiting for somebody to fix it [Tracking] ROCm packages · Issue #197885 · NixOS/nixpkgs · GitHub