nix-builder: update ROCm 7.2.1 hashes (#451)

* nix-builder: update ROCm 7.2.1 hashes

These changed a few days ago. Normally I'd be worried that it could be a
supply chain attack, but the repomd data is validly signed with the GPG
key of the ROCm team.

* nix-builder: more aggressive caching of ROCm packages

Author: danieldk

nix-builder/lib/extension/default.nix

@@ -56,7 +56,18 @@ in
       oneapi-torch-dev
       onednn-xpu
     ]
-    ++ lib.optionals rocmSupport [ clr ];
+    ++ lib.optionals rocmSupport (
+      [
+        clr
+      ]
+      ++ (with rocmPackages; [
+        hipcub-devel
+        hipsparselt
+        rocprim-devel
+        rocthrust-devel
+        rocwmma-devel
+      ])
+    );
 
   mkTvmFfiExtension = callPackage ./tvm-ffi/arch.nix {
     inherit