From 25dfa100da30e692d3d0f8cb666336eb8fe26e06 Mon Sep 17 00:00:00 2001
From: Tristan Ross <tristan.ross@determinate.systems>
Date: Fri, 5 Jun 2026 14:39:34 -0700
Subject: [PATCH] tt-smi: 3.0.30 -> 5.2.0

---
 pkgs/by-name/tt/tt-smi/package.nix | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/pkgs/by-name/tt/tt-smi/package.nix b/pkgs/by-name/tt/tt-smi/package.nix
index 9fced72b8bfd..912e59170ec4 100644
--- a/pkgs/by-name/tt/tt-smi/package.nix
+++ b/pkgs/by-name/tt/tt-smi/package.nix
@@ -4,17 +4,19 @@
   fetchFromGitHub,
   pre-commit,
   versionCheckHook,
+  tt-umd,
 }:
 python3Packages.buildPythonApplication (finalAttrs: {
   pname = "tt-smi";
-  version = "3.0.30";
+  version = "5.2.0";
   pyproject = true;
+  __structuredAttrs = true;
 
   src = fetchFromGitHub {
     owner = "tenstorrent";
     repo = "tt-smi";
     tag = "v${finalAttrs.version}";
-    hash = "sha256-C6CfcS0H3rFew/Y1uhmzICdFp1UYU7H9h3YPeAKlcbE=";
+    hash = "sha256-meDqvDvGBXx/zbHbtgLUb+Kv8LSmsu9OvYCFhmEPAdQ=";
   };
 
   build-system = with python3Packages; [
@@ -33,8 +35,11 @@ python3Packages.buildPythonApplication (finalAttrs: {
     tt-tools-common
     setuptools
     tomli
+    tt-umd
   ];
 
+  pythonRelaxDeps = [ "tt-umd" ];
+
   nativeCheckInputs = [
     versionCheckHook
   ];