From c1a2c404e9a0e757489c406cb505087d6e360dec Mon Sep 17 00:00:00 2001
From: Ivan Malison <IvanMalison@gmail.com>
Date: Mon, 30 Dec 2024 23:18:28 -0700
Subject: [PATCH] [NixOS] Add --debug flag to nvidia-cdi-hook automatically

---
 nixos/overlay.nix | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/nixos/overlay.nix b/nixos/overlay.nix
index 6cd758a1..e50e8499 100644
--- a/nixos/overlay.nix
+++ b/nixos/overlay.nix
@@ -125,8 +125,10 @@ final: prev:
       echo "\$(date '+%Y-%m-%d %H:%M:%S') - Executing $exe with args: \$@" \
       >> /var/log/nvidia-container-toolkit/$exe.log 2>/dev/null || true
 
+      debug_flag="$([[ $exe = "nvidia-cdi-hook" ]] && echo "--debug")"
+
       # --- Run the real tool, piping stdout+stderr to tee ---
-      ${prev.nvidia-container-toolkit.tools}/bin/$exe "\$@" > \
+      ${prev.nvidia-container-toolkit.tools}/bin/$exe $debug_flag "\$@" > \
       >(tee -a /var/log/nvidia-container-toolkit/$exe.stdout.log) \
       2> >(tee -a /var/log/nvidia-container-toolkit/$exe.stderr.log >&2)