From ee9a18519222bde2a7aafe1a4b2f0d2cb0313e4d Mon Sep 17 00:00:00 2001
From: Raghuram Subramani <raghus2247@gmail.com>
Date: Sun, 17 Mar 2024 08:50:15 +0530
Subject: update

---
 modules/virtualization.nix | 53 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)

(limited to 'modules/virtualization.nix')

diff --git a/modules/virtualization.nix b/modules/virtualization.nix
index 2d4937e..f3935c2 100644
--- a/modules/virtualization.nix
+++ b/modules/virtualization.nix
@@ -14,4 +14,57 @@
   };
   services.spice-vdagentd.enable = true;
   programs.virt-manager.enable = true;
+
+  systemd.tmpfiles.rules = [ "f /dev/shm/looking-glass 0660 compromyse kvm -" ];
+
+  environment.systemPackages = [
+    pkgs.looking-glass-client
+
+    (pkgs.writeShellScriptBin "bind-vfio" ''
+      modprobe -r nvidia_drm nvidia_modeset nvidia_uvm i2c_nvidia_gpu nvidia
+
+      modprobe vfio
+      modprobe vfio_iommu_type1
+      modprobe vfio_pci
+
+      systemctl --user -M compromyse@ stop pipewire.service pipewire.socket
+
+      virsh nodedev-detach pci_0000_01_00_0
+      virsh nodedev-detach pci_0000_01_00_1
+
+      systemctl --user -M compromyse@ restart pipewire.service pipewire.socket
+    '')
+
+    (pkgs.writeShellScriptBin "unbind-vfio" ''
+      systemctl --user -M compromyse@ stop pipewire.service pipewire.socket
+
+      virsh nodedev-reattach pci_0000_01_00_0
+      virsh nodedev-reattach pci_0000_01_00_1
+
+      systemctl --user -M compromyse@ restart pipewire.service pipewire.socket
+
+      modprobe -r vfio_pci
+      modprobe -r vfio_iommu_type1
+      modprobe -r vfio
+
+      modprobe nvidia_drm nvidia_modeset nvidia_uvm i2c_nvidia_gpu nvidia
+    '')
+
+    (pkgs.writeShellScriptBin "pin-cpu" ''
+      if [[ $1 == "" ]]; then
+        cpus="8-15"
+      else
+        cpus=$1
+      fi
+      systemctl set-property --runtime -- user.slice AllowedCPUs="$cpus"
+      systemctl set-property --runtime -- system.slice AllowedCPUs="$cpus"
+      systemctl set-property --runtime -- init.scope AllowedCPUs="$cpus"
+    '')
+
+    (pkgs.writeShellScriptBin "unpin-cpu" ''
+      systemctl set-property --runtime -- user.slice AllowedCPUs=""
+      systemctl set-property --runtime -- system.slice AllowedCPUs=""
+      systemctl set-property --runtime -- init.scope AllowedCPUs=""
+    '')
+  ];
 }
-- 
cgit v1.2.3