From e2dc9b3301301293958ebb5f5c7aa49c4da6c066 Mon Sep 17 00:00:00 2001 From: Sonia Sharma Date: Wed, 25 May 2022 15:10:44 -0700 Subject: [PATCH 1/2] features: gpu: update api-version which fixes gpu installation using extensions --- lisa/sut_orchestrator/azure/features.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lisa/sut_orchestrator/azure/features.py b/lisa/sut_orchestrator/azure/features.py index 9142f1e937..346f95036c 100644 --- a/lisa/sut_orchestrator/azure/features.py +++ b/lisa/sut_orchestrator/azure/features.py @@ -170,7 +170,7 @@ class Gpu(AzureFeatureMixin, features.Gpu): { "publisher": "Microsoft.HpcCompute", "type": "NvidiaGpuDriverLinux", - "typeHandlerVersion": "1.6", + "typeHandlerVersion": "1.7", "autoUpgradeMinorVersion": true, "settings": { } From d7a9c8a76785c8ee9d14d73ed38631c206d073b4 Mon Sep 17 00:00:00 2001 From: Sonia Sharma Date: Wed, 25 May 2022 15:27:56 -0700 Subject: [PATCH 2/2] features: gpu: update supportability for gpu drivers. --- lisa/sut_orchestrator/azure/features.py | 17 +++++++++++------ 1 file changed, 11 insertions(+), 6 deletions(-) diff --git a/lisa/sut_orchestrator/azure/features.py b/lisa/sut_orchestrator/azure/features.py index 346f95036c..4d38b1924d 100644 --- a/lisa/sut_orchestrator/azure/features.py +++ b/lisa/sut_orchestrator/azure/features.py @@ -28,7 +28,7 @@ from lisa.features.gpu import ComputeSDK from lisa.features.resize import ResizeAction from lisa.node import Node, RemoteNode -from lisa.operating_system import CentOs, Redhat, Suse, Ubuntu +from lisa.operating_system import Redhat, Suse, Ubuntu from lisa.search_space import RequirementMethod from lisa.tools import Dmesg, Lspci, Modprobe from lisa.util import ( @@ -180,11 +180,16 @@ class Gpu(AzureFeatureMixin, features.Gpu): def is_supported(self) -> bool: # TODO: more supportability can be defined here - supported_distro = (CentOs, Redhat, Ubuntu, Suse) - if isinstance(self._node.os, supported_distro): - return True - - return False + node = self._node + supported = False + if isinstance(node.os, Redhat): + supported = node.os.information.version >= "7.0.0" + elif isinstance(node.os, Ubuntu): + supported = node.os.information.version >= "18.0.0" + elif isinstance(node.os, Suse): + supported = node.os.information.version >= "15.0.0" + + return supported def get_supported_driver(self) -> List[ComputeSDK]: driver_list = []