From 2cad0340f1c9e4ad0019c7291d2ed5f89b963be4 Mon Sep 17 00:00:00 2001
From: Antoine Viallon <antoine@lesviallon.fr>
Date: Tue, 20 Feb 2024 00:51:34 +0100
Subject: [PATCH] [Hardware/Nvidia] help fix power management issues

---
 hardware/nvidia/proprietary.nix | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/hardware/nvidia/proprietary.nix b/hardware/nvidia/proprietary.nix
index 15ece0c..f2e03d6 100644
--- a/hardware/nvidia/proprietary.nix
+++ b/hardware/nvidia/proprietary.nix
@@ -94,10 +94,13 @@ in {
       // {
         "nvidia-drm.modeset" = 1;
         "nvidia.NVreg_UsePageAttributeTable" = 1;
+        "nvidia.NVreg_InitializeSystemMemoryAllocations" = 0;
       }
-      // {
-        "nvidia.NVreg_PreserveVideoMemoryAllocations" = toValue cfg.proprietary.saveAllVram;
-        "nvidia.NVreg_TemporaryFilePath" = "/var/tmp/nvidia-gpu.vram.img";
+      // optionalAttrs cfg.proprietary.saveAllVram {
+        # "nvidia.NVreg_PreserveVideoMemoryAllocations" = 1; # Already setby hardware.nvidia.powerManagement.enable
+        "nvidia.NVreg_DynamicPowerManagement" = "0x02";
+        "nvidia.NVreg_EnableS0ixPowerManagement" = 1;
+        "nvidia.NVreg_TemporaryFilePath" = "/var/tmp";
       }
     ;