umd: Update

pkgs/umd/default.nix

@@ -1,24 +1,73 @@
 { pkgs }:
 
+let
+  depsDir = "deps";
+
+  version = "unstable-2024-10-01";
+
+  # Update outputHash in umd-deps too
+  src = pkgs.fetchFromGitHub {
+    owner = "tenstorrent";
+    repo = "tt-umd";
+    rev = "5293e508ade90e758b386068770a882667a535f0";
+    hash = "sha256-Pa6vFjsG30UnGwLKZc/OTmiSMkXs/0sAFwVGojE8IbY=";
+  };
+
+  umd-deps = pkgs.stdenv.mkDerivation {
+    name = "tt-umd-deps-${version}.tar.gz";
+
+    inherit src;
+
+    dontBuild = true;
+
+    outputHash = "sha256-/T7UJ1OCd4T68DXOxHjGLYiiLoEZIRt7PHbT9npT4uk=";
+    outputHashAlgo = "sha256";
+
+    nativeBuildInputs = with pkgs; [
+      cmake
+      git
+      cacert
+      python3
+    ];
+
+    ARCH_NAME = "wormhole_b0";
+    cmakeFlags = [
+      "-DCPM_DOWNLOAD_ALL=ON"
+      "-DCPM_SOURCE_CACHE=${depsDir}"
+      "-DTT_UMD_BUILD_TESTS=ON"
+    ];
+
+    postPatch = ''
+      cp ${pkgs.cpm-cmake}/share/cpm/CPM.cmake cmake/CPM.cmake
+    '';
+
+    installPhase = ''
+      runHook preInstall
+
+      # Prune the `.git` directories
+      find ${depsDir} -name .git -type d -prune -exec rm -rf {} \;;
+      # Build a reproducible tar, per instructions at https://reproducible-builds.org/docs/archives/
+      tar --owner=0 --group=0 --numeric-owner --format=gnu \
+          --sort=name --mtime="@$SOURCE_DATE_EPOCH" \
+          -czf $out \
+            ${depsDir} \
+
+      runHook postInstall
+    '';
+  };
+in
+
 pkgs.stdenv.mkDerivation {
   pname = "tt-umd";
-  version = "main-2024-02-01";
+  inherit version src;
 
-  src = pkgs.fetchFromGitHub {
-    owner = "tenstorrent-metal";
-    repo = "umd";
-    rev = "341f5b7b299f128faaf2ca446a03298cb781a645";
-    hash = "sha256-jMxhhFWnCjNZZvFiTCeuEHvxvE0+IoaP4NJkr/CDLy8=";
-  };
-
-  patches = [
-    ./fmt_mystery.patch
-    ./missing_headers.patch
-  ];
-
-  makeFlags = [
-    "DEVICE_CXX=${pkgs.stdenv.cc.targetPrefix}c++"
-    "ARCH_NAME=grayskull"
+  nativeBuildInputs = with pkgs; [
+    cmake
+    git
+    cacert
+    ninja
+    python3
+    removeReferencesTo
   ];
 
   buildInputs = with pkgs; [
@@ -28,8 +77,28 @@ pkgs.stdenv.mkDerivation {
     hwloc
   ];
 
+  ARCH_NAME = "wormhole_b0";
+  cmakeFlags = [
+    "-DCPM_SOURCE_CACHE=${depsDir}"
+    # libdevice.so
+    # RUNPATH              /build/source/build/_deps/nanomsg-build:/build/source/build/_deps/libuv-build:/nix/store/n1yy5f1754p2d6dhksvg6rwpayymw1fx-tt-umd-unstable-2024-10-01/lib/:...
+    # TODO: look in to fixing properly if there's a need.
+    "-DCMAKE_SKIP_BUILD_RPATH=ON"
+  ];
+
+  postUnpack = ''
+    mkdir -p $sourceRoot/build
+    tar -xf ${umd-deps} -C $sourceRoot/build
+  '';
+
+  postPatch = ''
+    cp ${pkgs.cpm-cmake}/share/cpm/CPM.cmake cmake/CPM.cmake
+  '';
+
   installPhase = ''
-    mkdir $out
-    mv build/lib $out
+    runHook preInstall
+    mkdir -p $out/lib
+    mv lib/libdevice.so $out/lib
+    runHook postInstall
   '';
 }