wait for mofed driver installation after the cleanup driver call

Signed-off-by: Tariq Ibrahim <tibrahim@nvidia.com>
NVIDIA · Feb 21, 2024 · d64af46 · d64af46
1 parent 1461ff9
commit d64af46
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/driver-manager b/driver-manager
@@ -571,12 +571,6 @@ _wait_for_mofed_driver() {
 }
 
 uninstall_driver() {
-    # when GPUDirectRDMA is enabled, wait until MOFED driver has finished installing
-    if _gpu_direct_rdma_enabled; then
-        echo "GPUDirectRDMA is enabled, validating MOFED driver installation"
-        _wait_for_mofed_driver
-    fi
-
     # don't attempt to un-install if driver is pre-installed on the node
     if _host_driver; then
         echo "NVIDIA GPU driver is already pre-installed on the node, disabling the containerized driver on the node"
@@ -645,6 +639,12 @@ uninstall_driver() {
         _exit_failed
     fi
 
+    # when GPUDirectRDMA is enabled, wait until MOFED driver has finished installing
+    if _gpu_direct_rdma_enabled; then
+        echo "GPUDirectRDMA is enabled, validating MOFED driver installation"
+        _wait_for_mofed_driver
+    fi
+
     if _is_gpu_pod_eviction_enabled || _is_auto_drain_enabled; then
         # uncordon the node in case if the pod has restarted abruptly after we cordoned the node
         _uncordon_k8s_node