feat: add tenstorrent extension

.kres.yaml

@@ -49,6 +49,7 @@ spec:
     - spin
     - stargz-snapshotter
     - tailscale
+    - tenstorrent
     - thunderbolt
     - uinput
     - usb-modem-drivers

Makefile

@@ -104,6 +104,7 @@ TARGETS += revpi-firmware
 TARGETS += spin
 TARGETS += stargz-snapshotter
 TARGETS += tailscale
+TARGETS += tenstorrent
 TARGETS += thunderbolt
 TARGETS += uinput
 TARGETS += usb-modem-drivers

Pkgfile

@@ -13,6 +13,7 @@ vars:
   UTIL_LINUX_SHA256: 5c1daf733b04e9859afdc3bd87cc481180ee0f88b5c0946b16fdec931975fb79
   UTIL_LINUX_SHA512: 0024955056ba7b4c54040a917f9919f49692e57ba6d42d17a6c29c1eefe88bf48b1214a545072b71c468829a63a8f15237f49733e9127c134e11126d1e435124
   ENA_VERSION: 2.13.3 # update this when updating PKGS_VERSION in Makefile
+  TENSTORRENT_VERSION: 1.33
 
   # renovate: datasource=git-tags extractVersion=^libtiprc-(?<version>.*)$ depName=git://linux-nfs.org/~steved/libtirpc
   LIBTIRPC_VERSION: 1-3-3

drivers/tenstorrent/README.md

@@ -0,0 +1,35 @@
+# tenstorrent extension
+
+This extension provides [Tennstorrent](https://github.com/tenstorrent/tt-kmd) drivers for Talos Linux.
+
+## Installation
+
+See [Installing Extensions](https://github.com/siderolabs/extensions#installing-extensions).
+
+## Usage
+
+```yaml
+machine:
+  kernel:
+    modules:
+      - name: tenstorrent
+```
+
+## Verifying
+
+You can verify the modules are enabled by reading the `/proc/modules` where it _should_ show the module is live.
+
+For example:
+
+```bash
+❯ talosctl read /proc/modules
+```
+
+In addition, if you actually have Tenstorrent hardware installed, you should be able to verify the devices that begin with `/dev/tenstorrent/*`.
+
+For example:
+
+```bash
+❯ talosctl ls -l /dev/tenstorrent/*
+NODE           MODE          UID   GID   SIZE(B)   LASTMOD           NAME
+```

drivers/tenstorrent/manifest.yaml

@@ -0,0 +1,10 @@
+version: v1alpha1
+metadata:
+  name: tenstorrent
+  version: "$VERSION"
+  author: Tenstorrent AI ULC
+  description: |
+    Driver for Tenstorrent AI processing hardware
+  compatibility:
+    talos:
+      version: ">= v1.9.0"

drivers/tenstorrent/pkg.yaml

@@ -0,0 +1,25 @@
+name: tenstorrent
+variant: scratch
+shell: /bin/bash
+dependencies:
+  - stage: base
+  # The pkgs version for a particular release of Talos as defined in
+  # https://github.com/siderolabs/talos/blob/<talos version>/pkg/machinery/gendata/data/pkgs
+  - image: "{{ .BUILD_ARG_PKGS_PREFIX }}/tenstorrent-pkg:{{ .VERSION }}"
+steps:
+  - prepare:
+      - |
+        sed -i 's#$VERSION#{{ .VERSION }}#' /pkg/manifest.yaml
+  - install:
+      - |
+        mkdir -p /rootfs/usr/lib/modules
+        mkdir -p /rootfs/etc/udev/rules.d
+
+      - |
+        cp -R /usr/lib/modules/* /rootfs/usr/lib/modules/
+        cp -R /etc/udev/rules.d/* /rootfs/etc/udev/rules.d/
+finalize:
+  - from: /rootfs
+    to: /rootfs
+  - from: /pkg/manifest.yaml
+    to: /

drivers/tenstorrent/vars.yaml

@@ -0,0 +1,2 @@
+# update when pkgs version is updated
+VERSION: "{{ .TENSTORRENT_VERSION }}"