Back to prior_to file_line

ComputeCanada · Jan 23, 2025 · b72faf6 · b72faf6
1 parent d17e8fd
commit b72faf6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/site/profile/manifests/gpu.pp b/site/profile/manifests/gpu.pp
@@ -47,18 +47,18 @@
     transition { 'stop nvidia-dcgm service':
       resource   => Service['nvidia-dcgm'],
       attributes => { ensure => stopped },
-      prior_to   => Exec['unload nvidia drivers'],
+      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'],
+      prior_to   => File_line['nvidia_restrict_profiling'],
     }
   } else {
     transition { 'stop nvidia-gridd service':
       resource   => Service['nvidia-gridd'],
       attributes => { ensure => stopped },
-      prior_to   => Exec['unload nvidia drivers'],
+      prior_to   => File_line['nvidia_restrict_profiling'],
     }
   }