pyrox.dev
1--- a/setup.py
2+++ b/setup.py
3@@ -340,14 +340,7 @@ def _is_hpu() -> bool:
4 out = subprocess.run(["hl-smi"], capture_output=True, check=True)
5 is_hpu_available = out.returncode == 0
6 except (FileNotFoundError, PermissionError, subprocess.CalledProcessError):
7- if sys.platform.startswith("linux"):
8- try:
9- output = subprocess.check_output(
10- 'lsmod | grep habanalabs | wc -l', shell=True)
11- is_hpu_available = int(output) > 0
12- except (ValueError, FileNotFoundError, PermissionError,
13- subprocess.CalledProcessError):
14- pass
15+ is_hpu_available = False
16 return is_hpu_available
17
18