Fix (gptq): linalg import fix

Xilinx · Sep 12, 2023 · 1587c17 · 1587c17
1 parent 461a0e0
commit 1587c17
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/brevitas/graph/gptq.py b/src/brevitas/graph/gptq.py
@@ -11,7 +11,12 @@
 import warnings
 
 import torch
-from torch.linalg import LinAlgError
+
+try:
+    from torch.linalg import LinAlgError
+except:
+    LinAlgError = RuntimeError
+
 import unfoldNd
 
 from brevitas.graph.calibrate import DisableEnableQuantization