Change transition prior

ComputeCanada · Jan 23, 2025 · d17e8fd · d17e8fd
1 parent 896048d
commit d17e8fd
Showing 1 changed file with 9 additions and 13 deletions.
diff --git a/site/profile/manifests/gpu.pp b/site/profile/manifests/gpu.pp
@@ -44,25 +44,21 @@
   }
 
   if ! $facts['nvidia_grid_vgpu'] {
-    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-dcgm service':
+      resource   => Service['nvidia-dcgm'],
+      attributes => { ensure => stopped },
+      prior_to   => Exec['unload nvidia drivers'],
     }
-    if defined(Service['nvidia-dcgm']) {
-      transition { 'stop nvidia-persistenced service':
-        resource   => Service['nvidia-persistenced'],
-        attributes => { ensure => stopped },
-        prior_to   => File_line['nvidia_restrict_profiling'],
-      }
+    transition { 'stop nvidia-persistenced service':
+      resource   => Service['nvidia-persistenced'],
+      attributes => { ensure => stopped },
+      prior_to   => Exec['unload nvidia drivers'],
     }
   } else {
     transition { 'stop nvidia-gridd service':
       resource   => Service['nvidia-gridd'],
       attributes => { ensure => stopped },
-      prior_to   => File_line['nvidia_restrict_profiling'],
+      prior_to   => Exec['unload nvidia drivers'],
     }
   }