Init

pkgs/kmd/default.nix

@@ -0,0 +1,59 @@
+{ pkgs, kernel }:
+
+let
+  src = pkgs.fetchFromGitHub {
+    owner = "tenstorrent";
+    repo = "tt-kmd";
+    rev = "455b948";
+    hash = "sha256-oJzAPd85Cyd1hcuFPJNyHA/YXX89fTUK5o4a+M4Nu00=";
+  };
+  version = "main-01-31-24";
+in
+{
+  kmd = pkgs.stdenv.mkDerivation {
+    pname = "tt-kmd";
+
+    inherit src version;
+
+    nativeBuildInputs = kernel.moduleBuildDependencies;
+    hardeningDisable = [ "all" ];
+    buildPhase = ''
+      make modules -C ${kernel.dev}/lib/modules/${kernel.modDirVersion}/build M=$(pwd -P)
+    '';
+    installPhase = ''
+      mkdir -p $out/lib/modules/${kernel.modDirVersion}/extra
+      cp tenstorrent.ko $out/lib/modules/${kernel.modDirVersion}/extra/
+    '';
+  };
+
+  udev-rules = pkgs.stdenv.mkDerivation rec {
+    pname = "tenstorrent-udev-rules";
+
+    inherit src version;
+
+    dontUnpack = true;
+
+    installPhase = ''
+      install -Dpm644 $src/udev-50-tenstorrent.rules $out/lib/udev/rules.d/50-tenstorrent.rules
+    '';
+  };
+
+  test = pkgs.stdenv.mkDerivation {
+    pname = "tt-kmd-test";
+
+    inherit src version;
+
+    patches = [ ./missing_header.patch ];
+
+    nativeBuildInputs = [ pkgs.gnumake ];
+
+    buildPhase = ''
+      make -C test
+    '';
+
+    installPhase = ''
+      mkdir -p $out/bin
+      cp test/ttkmd_test $out/bin/tt-kmd-test
+    '';
+  };
+}

pkgs/kmd/missing_header.patch

@@ -0,0 +1,24 @@
+diff --git a/test/dma_buf.cpp b/test/dma_buf.cpp
+index a34bc44..95718a1 100644
+--- a/test/dma_buf.cpp
++++ b/test/dma_buf.cpp
+@@ -4,6 +4,7 @@
+ #include <limits>
+ #include <variant>
+ #include <cerrno>
++#include <cstdint>
+ 
+ #include <sys/ioctl.h>
+ #include <sys/mman.h>
+diff --git a/test/lock.cpp b/test/lock.cpp
+index 941006e..0272126 100644
+--- a/test/lock.cpp
++++ b/test/lock.cpp
+@@ -3,6 +3,7 @@
+ 
+ #include <optional>
+ #include <string>
++#include <cstdint>
+ 
+ #include <errno.h>
+ #include <sys/ioctl.h>