From 9ff1720c53cca5f61fd615b30b2e6893a35a0596 Mon Sep 17 00:00:00 2001 From: Vidya S Galli Date: Mon, 5 Feb 2024 19:34:00 +0000 Subject: [PATCH 1/2] update version check --- optimum/habana/utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/optimum/habana/utils.py b/optimum/habana/utils.py index 6a92a42036..ad67874039 100644 --- a/optimum/habana/utils.py +++ b/optimum/habana/utils.py @@ -222,14 +222,16 @@ def get_driver_version(): """ Returns the driver version. """ + # Enable console printing for `hl-smi` check output = subprocess.run( "hl-smi", shell=True, text=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, + env={"ENABLE_CONSOLE": "true"} ) - if output.returncode == 0: + if output.returncode == 0 and output.stdout: return version.parse(output.stdout.split("\n")[2].replace(" ", "").split(":")[1][:-1].split("-")[0]) return None From f065b868ab2820465e42dfedcfee5e3632594fc3 Mon Sep 17 00:00:00 2001 From: Vidya Galli Date: Tue, 6 Feb 2024 09:26:31 -0800 Subject: [PATCH 2/2] fixed formatting --- optimum/habana/utils.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/optimum/habana/utils.py b/optimum/habana/utils.py index ad67874039..fb4541e5d9 100644 --- a/optimum/habana/utils.py +++ b/optimum/habana/utils.py @@ -224,12 +224,7 @@ def get_driver_version(): """ # Enable console printing for `hl-smi` check output = subprocess.run( - "hl-smi", - shell=True, - text=True, - stdout=subprocess.PIPE, - stderr=subprocess.PIPE, - env={"ENABLE_CONSOLE": "true"} + "hl-smi", shell=True, text=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, env={"ENABLE_CONSOLE": "true"} ) if output.returncode == 0 and output.stdout: return version.parse(output.stdout.split("\n")[2].replace(" ", "").split(":")[1][:-1].split("-")[0])