Detect amdgpu module for amdgpu-builtin kernel From dc2a4b5b256692e1a6ecef45210682c316b961e0 Mon Sep 17 00:00:00 2001 From: YiyangWu Date: Sat, 2 Apr 2022 16:40:11 +0800 Subject: [PATCH] Check /sys/module/amdgpu for ROCk instead of lsmod Closes: #102 Signed-off-by: YiyangWu --- python_smi_tools/rocm_smi.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) Index: rocm_smi_lib-rocm-5.4.2/python_smi_tools/rocm_smi.py =================================================================== --- rocm_smi_lib-rocm-5.4.2.orig/python_smi_tools/rocm_smi.py +++ rocm_smi_lib-rocm-5.4.2/python_smi_tools/rocm_smi.py @@ -66,7 +66,7 @@ def driverInitialized(): """ driverInitialized = '' try: - driverInitialized = str(subprocess.check_output("cat /sys/module/amdgpu/initstate |grep live", shell=True)) + driverInitialized = str(subprocess.check_output("test -d /sys/module/amdgpu", shell=True)) except subprocess.CalledProcessError: pass if len(driverInitialized) > 0: