rocmPackages.aotriton: 0.9.2b -> 0.10b

[LunNova]

Diff: ROCm/aotriton@0.9.2b...0.10b

This is a rewrite of aotriton/default.nix. We're properly reusing the existing python3Packages.triton instance as the build-time triton instead of letting aotriton's CMakeLists.txt build its own triton, so lots of hackery can be dropped! and some packages that weren't actually used and were in buildInputs 😅

If in future we needed to pin a specific triton source rev here with an override it should be relatively safe as it's a build time only dependency and won't run into collisions like in many cases in python packaging.
For now we have no need to pin triton; aotriton builds happily against triton 3.4.

This is a followup to the recent torch bump #431973. torch 2.8 introduced faster sliding window attention via aotriton but requires a minimum version of 0.10b to enable it.

https://github.com/pytorch/pytorch/blob/379ebdaf5e2fbf387de297f673254d005e583b31/aten/src/ATen/native/transformers/hip/flash_attn/aot/mha_all_aot.hip#L96-L119

Things done

• Built on platform:
  • x86_64-linux
  • aarch64-linux
  • x86_64-darwin
  • aarch64-darwin
• Tested, as applicable:
  • NixOS tests in nixos/tests.
  • Package tests at passthru.tests.
  • Tests in lib/tests or pkgs/test for functions and "core" functionality.
• Ran nixpkgs-review on this PR. See nixpkgs-review usage.
• Tested basic functionality of all binary files, usually in ./result/bin/.
• Nixpkgs Release Notes
  • Package update: when the change is major or breaking.
• NixOS Release Notes
  • Module addition: when adding a new NixOS module.
  • Module update: when the change is significant.
• Fits CONTRIBUTING.md, pkgs/README.md, maintainers/README.md and other READMEs.

---

Add a 👍 reaction to pull requests you find important.

[Flakebi]

Thanks, LGTM!

[nixos-discourse]

This pull request has been mentioned on NixOS Discourse. There might be relevant details there:

https://discourse.nixos.org/t/prs-already-reviewed/2617/2530

[pbsds]

I like the new expression as it reads a lot better. My build is still running, but some comments for now

[LunNova]

Oh that's fun. The combination of this diff and 2a2a51e doesn't build, but either works alone. Marking draft since that just got merged, glad we didn't merge both in quick succession.

[LunNova]

nixpkgs-review result

Generated using nixpkgs-review.

Command: nixpkgs-review pr 439056
Commit: 400c14cf497caf497319d46682eaceee8e80d821

---

x86_64-linux

✅ 14 packages built:

• python312Packages.torchWithRocm
• python312Packages.torchWithRocm.cxxdev
• python312Packages.torchWithRocm.dev
• python312Packages.torchWithRocm.dist
• python312Packages.torchWithRocm.lib
• python313Packages.torchWithRocm
• python313Packages.torchWithRocm.cxxdev
• python313Packages.torchWithRocm.dev
• python313Packages.torchWithRocm.dist
• python313Packages.torchWithRocm.lib
• rocmPackages.aotriton (rocmPackages_6.aotriton)
• rocmPackages.aotriton.debug (rocmPackages_6.aotriton.debug)
• rocmPackages.mivisionx (rocmPackages.mivisionx-hip, rocmPackages_6.mivisionx, rocmPackages_6.mivisionx-hip)
• rocmPackages.mivisionx-cpu (rocmPackages_6.mivisionx-cpu)

[GaetanLepage]

LGTM overall. Made minor comments.

[LunNova]

Applied both suggestions

[GaetanLepage]

Thanks. Would like a second opinion though.