Fix (fx): import backport on 2.1 (#732)

Xilinx · Oct 20, 2023 · 48a24bd · 48a24bd
1 parent 1412174
commit 48a24bd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/brevitas/fx/__init__.py b/src/brevitas/fx/__init__.py
@@ -7,7 +7,7 @@
 from brevitas import torch_version
 
 # This needs to be bumped until https://github.com/pytorch/pytorch/pull/94461 gets merged
-if torch_version < version.parse('2.1'):
+if torch_version < version.parse('2.2'):
     import brevitas.backport.fx as fx
     from brevitas.backport.fx._compatibility import compatibility
     from brevitas.backport.fx._symbolic_trace import _assert_is_none