Fix (gptq): linalg import fix (#705)

Xilinx · Sep 12, 2023 · 102b7f8 · 102b7f8
1 parent 2b203c9
commit 102b7f8
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