feat(krz01): passthrough the nVidia Tesla K80 in ml01

This way, no need for reboot.

Signed-off-by: Ryan Lahfa <ryan@dgnum.eu>

machines/krz01/_configuration.nix

@@ -15,7 +15,9 @@ lib.extra.mkConfig {
   ];
 
   extraConfig = {
-    microvm.host.enable = true;
+    microvm = {
+      host.enable = true;
+    };
     dgn-hardware = {
       useZfs = true;
       zfsPools = [

machines/krz01/microvm-ml01.nix

@@ -1,7 +1,15 @@
 _: {
   microvm.autostart = [ "ml01" ];
   microvm.vms.ml01 = {
-    config = {
+    config =
+      { config, ... }:
+      {
+        nixpkgs.config.cudaSupport = true;
+        nixpkgs.config.nvidia.acceptLicense = true;
+        # Tesla K80 is not supported by the latest driver.
+        hardware.nvidia.package = config.boot.kernelPackages.nvidia_x11_legacy470;
+        # Don't ask.
+        services.xserver.videoDrivers = [ "nvidia" ];
         networking.hostName = "ml01";
         services.ollama = {
           enable = true;
@@ -9,7 +17,24 @@ _: {
           sandbox = true;
           acceleration = "cuda";
         };
-      microvm.shares = [
+
+        microvm = {
+          hypervisor = "cloud-hypervisor";
+          vcpu = 4;
+          mem = 4096;
+          balloonMem = 2048;
+          devices = [
+            # The nVidia Tesla K80
+            {
+              bus = "pci";
+              path = "0000:44:00.0";
+            }
+            {
+              bus = "pci";
+              path = "0000:45:00.0";
+            }
+          ];
+          shares = [
             {
               source = "/nix/store";
               mountPoint = "/nix/.ro-store";
@@ -19,4 +44,5 @@ _: {
           ];
         };
       };
+  };
 }