pyrox.dev
1{
2 clr,
3 ollama,
4 python3Packages,
5 rocmPackages,
6 magma-hip,
7 emptyDirectory,
8 stdenv,
9}:
10# This package exists purely to have a bunch of passthru.tests attrs
11stdenv.mkDerivation {
12 name = "rocm-tests";
13 nativeBuildInputs = [
14 clr
15 ];
16 src = emptyDirectory;
17 postInstall = "mkdir -p $out";
18 passthru.tests = {
19 ollama = ollama.override {
20 inherit rocmPackages;
21 acceleration = "rocm";
22 };
23 torch = python3Packages.torch.override {
24 inherit rocmPackages;
25 rocmSupport = true;
26 cudaSupport = false;
27 magma-hip = magma-hip.override {
28 inherit rocmPackages;
29 };
30 };
31 };
32}