diff --git a/l4casadi/l4casadi.py b/l4casadi/l4casadi.py index 114acdd..a4497c6 100644 --- a/l4casadi/l4casadi.py +++ b/l4casadi/l4casadi.py @@ -75,6 +75,7 @@ def __init__(self, if not scripting: warnings.warn("L4CasADi with Torch AOT compilation is experimental at this point and might not work as " "expected.") + raise RuntimeError("PyTorch compile is not supported yet as it does not seem stable.") if torch.__version__ < torch.torch_version.TorchVersion('2.4.0'): raise RuntimeError("For PyTorch versions < 2.4.0 L4CasADi only supports jit scripting. Please pass " "scripting=True.") @@ -386,7 +387,7 @@ def export_torch_traces(self) -> Tuple[bool, bool, bool, bool]: out_folder = self.build_dir - self.model_compile( make_fx(functionalize(self.model, remove='mutations_and_views'))(d_inp), + self.model_compile(make_fx(functionalize(self.model, remove='mutations_and_views'))(d_inp), (out_folder / f'{self.name}.pt').as_posix(), (d_inp,)) diff --git a/libl4casadi/CMakeLists.txt b/libl4casadi/CMakeLists.txt index 9229efa..674d250 100644 --- a/libl4casadi/CMakeLists.txt +++ b/libl4casadi/CMakeLists.txt @@ -22,7 +22,7 @@ add_definitions(-DTORCH_VERSION_MINOR=${Torch_VERSION_MINOR}) add_definitions(-DTORCH_VERSION_PATCH=${Torch_VERSION_PATCH}) if (Torch_VERSION_MAJOR GREATER_EQUAL 1 AND Torch_VERSION_MINOR GREATER_EQUAL 4) - #add_definitions(-DENABLE_TORCH_COMPILE) + # add_definitions(-DENABLE_TORCH_COMPILE) endif () if (USE_CUDA) add_definitions(-DUSE_CUDA) diff --git a/pyproject.toml b/pyproject.toml index 04d70b7..9b903d8 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -9,7 +9,7 @@ build-backend = "setuptools.build_meta" [project] name = "l4casadi" -version = "1.4.1" +version = "2.0.0" authors = [ { name="Tim Salzmann", email="Tim.Salzmann@tum.de" }, ]