alex
0813f55762
CI / check (push) Has been cancelled
Allow a pinned JIT SDK without rebuilding PyTorch or the extension. Keep runtime device bitcode unchanged: exporting TheRock bitcode into the serving process crashes CLR 7.2 on a basic tensor operation. Disable unsupported expandable segments and check all five HIP kernel modules, with an optional GPU matmul and module-loading regression test.
14 lines
516 B
Nix
14 lines
516 B
Nix
{
|
|
pkgs,
|
|
inputs,
|
|
jitRocmSdk ? null,
|
|
...
|
|
}:
|
|
# The HIP extension has to be compiled against a ROCm build of PyTorch, and the
|
|
# only ROCm torch in this flake with a cached binary is the one from the pinned
|
|
# nixpkgs-torch211 input (2.11 + ROCm 7.2, python 3.13). Current nixos-unstable
|
|
# ships torch 2.13, whose ROCm build fails to configure in nixpkgs and is not
|
|
# on cache.nixos.org.
|
|
inputs.nixpkgs-torch211.legacyPackages.${pkgs.stdenv.hostPlatform.system}.callPackage ./package.nix
|
|
{ inherit jitRocmSdk; }
|