From 81b6f4738235a29713aa01419fccc45ce8307907 Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Sat, 23 Nov 2024 20:48:39 +0000 Subject: [PATCH] Bump value tracer compatibility --- src/brevitas/fx/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/brevitas/fx/__init__.py b/src/brevitas/fx/__init__.py index 50e66a159..325f2dea4 100644 --- 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