From 7d33297fa84047e24b679586a089349c641130dc Mon Sep 17 00:00:00 2001 From: Lili Deng Date: Sat, 28 May 2022 20:35:24 +0800 Subject: [PATCH] Stop service which uses nvidia module, then remove module can succeed. --- microsoft/testsuites/gpu/gpusuite.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/microsoft/testsuites/gpu/gpusuite.py b/microsoft/testsuites/gpu/gpusuite.py index 7847bcfe99..c062fc4b12 100644 --- a/microsoft/testsuites/gpu/gpusuite.py +++ b/microsoft/testsuites/gpu/gpusuite.py @@ -20,7 +20,7 @@ from lisa.features import Gpu, GpuEnabled, SerialConsole from lisa.features.gpu import ComputeSDK from lisa.operating_system import Debian -from lisa.tools import Lspci, NvidiaSmi, Pip, Python, Reboot, Tar, Wget +from lisa.tools import Lspci, NvidiaSmi, Pip, Python, Reboot, Service, Tar, Wget from lisa.util import get_matched_str _cudnn_location = ( @@ -131,6 +131,9 @@ def verify_gpu_rescind_validation(self, node: Node) -> None: # 1. Disable GPU devices. gpu_devices = lspci.get_devices_by_type(device_type=constants.DEVICE_TYPE_GPU) gpu_devices = gpu.remove_virtual_gpus(gpu_devices) + # stop the service which uses nvidia module + service = node.tools[Service] + service.stop_service("nvidia-persistenced") for device in gpu_devices: lspci.disable_device(device)