From 5fece20512d07d35392a59c0cd065a79f530f858 Mon Sep 17 00:00:00 2001
From: KeithMnemonic <kberger@nvidia.com>
Date: Thu, 5 Sep 2024 11:54:10 -0400
Subject: [PATCH] Update nvidia-gpu-reset.target setup

This PR fixes an incorrect entry in the
nvidia-mig-manager.service. The "WantedBy" needs
to point to multi-uer.targer otherwise unless
something "Requires" or "Wants" nvidia-mig-manager
it will not ever be started.

Also the original patch that added the target did not
update the debian Makefile and RPM spec file. This
resulted in the target not being installed.

Signed-off-by: KeithMnemonic <kberger@nvidia.com>
---
 deployments/systemd/nvidia-mig-manager.service               | 2 +-
 deployments/systemd/packages/debian/Makefile                 | 2 ++
 .../systemd/packages/rpm/SPECS/nvidia-mig-manager.spec       | 5 ++++-
 3 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/deployments/systemd/nvidia-mig-manager.service b/deployments/systemd/nvidia-mig-manager.service
index e77773e4..6c4226ab 100644
--- a/deployments/systemd/nvidia-mig-manager.service
+++ b/deployments/systemd/nvidia-mig-manager.service
@@ -23,4 +23,4 @@ Type=oneshot
 ExecStart=-/bin/bash /etc/nvidia-mig-manager/service.sh
 
 [Install]
-WantedBy=nvidia-gpu-reset.target
+WantedBy=multi-user.target
diff --git a/deployments/systemd/packages/debian/Makefile b/deployments/systemd/packages/debian/Makefile
index cb087fce..e3b7acfe 100644
--- a/deployments/systemd/packages/debian/Makefile
+++ b/deployments/systemd/packages/debian/Makefile
@@ -24,6 +24,7 @@ SOURCE7 = hooks.sh
 SOURCE8 = hooks-default.yaml
 SOURCE9 = hooks-minimal.yaml
 SOURCE10 = config-default.yaml
+SOURCE11 = nvidia-gpu-reset.target
 
 build:
 
@@ -45,3 +46,4 @@ install:
 	install -m 644 -t $(DESTDIR)/etc/nvidia-mig-manager $(SOURCE8)
 	install -m 644 -t $(DESTDIR)/etc/nvidia-mig-manager $(SOURCE9)
 	install -m 644 -t $(DESTDIR)/etc/nvidia-mig-manager $(SOURCE10)
+	install -m 644 -t $(DESTDIR)/lib/systemd/system $(SOURCE11)
diff --git a/deployments/systemd/packages/rpm/SPECS/nvidia-mig-manager.spec b/deployments/systemd/packages/rpm/SPECS/nvidia-mig-manager.spec
index cdffc8e2..88a11194 100644
--- a/deployments/systemd/packages/rpm/SPECS/nvidia-mig-manager.spec
+++ b/deployments/systemd/packages/rpm/SPECS/nvidia-mig-manager.spec
@@ -19,6 +19,7 @@ Source7: hooks.sh
 Source8: hooks-default.yaml
 Source9: hooks-minimal.yaml
 Source10: config-default.yaml
+Source11: nvidia-gpu-reset.target
 
 %description
 The NVIDIA MIG Partition Editor allows administrators to declaratively define a
@@ -38,7 +39,7 @@ cp %{SOURCE0} %{SOURCE1} \
    %{SOURCE4} %{SOURCE5} \
    %{SOURCE6} %{SOURCE7} \
    %{SOURCE8} %{SOURCE9} \
-   %{SOURCE10} \
+   %{SOURCE10} %{SOURCE11} \
     .
 
 %install
@@ -59,6 +60,7 @@ install -m 644 -t %{buildroot}/etc/nvidia-mig-manager %{SOURCE7}
 install -m 644 -t %{buildroot}/etc/nvidia-mig-manager %{SOURCE8}
 install -m 644 -t %{buildroot}/etc/nvidia-mig-manager %{SOURCE9}
 install -m 644 -t %{buildroot}/etc/nvidia-mig-manager %{SOURCE10}
+install -m 644 -t %{buildroot}/usr/lib/systemd/system %{SOURCE11}
 
 %files
 %license LICENSE
@@ -75,6 +77,7 @@ install -m 644 -t %{buildroot}/etc/nvidia-mig-manager %{SOURCE10}
 %dir /etc/systemd/system/nvidia-mig-manager.service.d
 %dir /etc/nvidia-mig-manager/
 %dir /var/lib/nvidia-mig-manager
+/usr/lib/systemd/system/nvidia-gpu-reset.target
 
 %post
 systemctl daemon-reload