summaryrefslogtreecommitdiff
path: root/azurelinuxagent/pa/rdma
diff options
context:
space:
mode:
Diffstat (limited to 'azurelinuxagent/pa/rdma')
-rw-r--r--azurelinuxagent/pa/rdma/centos.py3
-rw-r--r--azurelinuxagent/pa/rdma/suse.py6
2 files changed, 3 insertions, 6 deletions
diff --git a/azurelinuxagent/pa/rdma/centos.py b/azurelinuxagent/pa/rdma/centos.py
index 8ad09c5..214f9ea 100644
--- a/azurelinuxagent/pa/rdma/centos.py
+++ b/azurelinuxagent/pa/rdma/centos.py
@@ -173,8 +173,7 @@ class CentOSRDMAHandler(RDMAHandler):
'user mode', self.rdma_user_mode_package_name, umod_pkg_path)
logger.info("RDMA: driver packages installed")
- self.load_driver_module()
- if not self.is_driver_loaded():
+ if not self.load_driver_module() or not self.is_driver_loaded():
logger.info("RDMA: driver module is not loaded; reboot required")
self.reboot_system()
else:
diff --git a/azurelinuxagent/pa/rdma/suse.py b/azurelinuxagent/pa/rdma/suse.py
index f0d8d0f..d31b2b0 100644
--- a/azurelinuxagent/pa/rdma/suse.py
+++ b/azurelinuxagent/pa/rdma/suse.py
@@ -93,8 +93,7 @@ class SUSERDMAHandler(RDMAHandler):
msg = 'RDMA: Successfully installed "%s" from '
msg += 'configured repositories'
logger.info(msg % complete_name)
- self.load_driver_module()
- if requires_reboot:
+ if not self.load_driver_module() or requires_reboot:
self.reboot_system()
return True
else:
@@ -119,8 +118,7 @@ class SUSERDMAHandler(RDMAHandler):
msg = 'RDMA: Successfully installed "%s" from '
msg += 'local package cache'
logger.info(msg % (local_package))
- self.load_driver_module()
- if requires_reboot:
+ if not self.load_driver_module() or requires_reboot:
self.reboot_system()
return True
else: