Execute transition only if service is defined

ComputeCanada · Jan 23, 2025 · 896048d · 896048d
1 parent 261e495
commit 896048d
Showing 1 changed file with 12 additions and 8 deletions.
diff --git a/site/profile/manifests/gpu.pp b/site/profile/manifests/gpu.pp
@@ -44,15 +44,19 @@
   }
 
   if ! $facts['nvidia_grid_vgpu'] {
-    transition { 'stop nvidia-dcgm service':
-      resource   => Service['nvidia-dcgm'],
-      attributes => { ensure => stopped },
-      prior_to   => File_line['nvidia_restrict_profiling'],
+    if defined(Service['nvidia-dcgm']) {
+      transition { 'stop nvidia-dcgm service':
+        resource   => Service['nvidia-dcgm'],
+        attributes => { ensure => stopped },
+        prior_to   => File_line['nvidia_restrict_profiling'],
+      }
     }
-    transition { 'stop nvidia-persistenced service':
-      resource   => Service['nvidia-persistenced'],
-      attributes => { ensure => stopped },
-      prior_to   => File_line['nvidia_restrict_profiling'],
+    if defined(Service['nvidia-dcgm']) {
+      transition { 'stop nvidia-persistenced service':
+        resource   => Service['nvidia-persistenced'],
+        attributes => { ensure => stopped },
+        prior_to   => File_line['nvidia_restrict_profiling'],
+      }
     }
   } else {
     transition { 'stop nvidia-gridd service':