Bump value tracer compatibility

Xilinx · Nov 23, 2024 · 81b6f47 · 81b6f47
1 parent 9f736cb
commit 81b6f47
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.2'):
+if torch_version < version.parse('2.5'):
     import brevitas.backport.fx as fx
     from brevitas.backport.fx._compatibility import compatibility
     from brevitas.backport.fx._symbolic_trace import _assert_is_none