Merge pull request #24 from mila-iqia/feat/mlnx_ofed_flavor

feat: Install mlnx-ofed-basic by default
mila-iqia · Apr 9, 2024 · cf356d9 · cf356d9
2 parents 753cc6a + 0df9433
commit cf356d9
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/defaults/main.yml b/defaults/main.yml
@@ -38,6 +38,11 @@ infiniband_upgrade: False
 # Name of the kernel headers package
 infiniband_kernel_headers_package: 'linux-headers'
 
+# Flavor of mlnx-ofed-* to install.
+# Values can be `basic`, `all`, `hpc`, etc.
+# See NVIDIA MLNX_OFED documentation for a complete list of flavors.
+infiniband_mlnx_ofed_flavor: 'basic'
+
 # Default to configure the repos and install the kernel modules.
 #
 # Some appliances like the NVIDIA DGX run their own software stack. When an

diff --git a/tasks/install-Debian.yml b/tasks/install-Debian.yml
@@ -2,8 +2,7 @@
 - name: Install MLNX_OFED packages for Debian OS family
   ansible.builtin.package:
     name:
-      - mlnx-ofed-kernel-dkms
-      - mlnx-ofed-kernel-utils
+      - "mlnx-ofed-{{ infiniband_mlnx_ofed_flavor }}"
       - "{{ infiniband_kernel_headers_package }}-{{ ansible_facts['kernel'] }}"
     state: "{{ 'latest' if infiniband_upgrade else 'present' }}"
   register: packages_install

diff --git a/tasks/install-RedHat.yml b/tasks/install-RedHat.yml
@@ -2,7 +2,6 @@
 - name: Install MLNX_OFED packages for Red Hat OS family
   ansible.builtin.package:
     name:
-      - mlnx-ofa_kernel-devel
-      - kmod-mlnx-ofa_kernel
+      - "mlnx-ofed-{{ infiniband_mlnx_ofed_flavor }}"
     state: "{{ 'latest' if infiniband_upgrade else 'present' }}"
   notify: "Ensure openibd.service is started"