Merge branch 'develop' into develop

Author: kratman

CHANGELOG.md

@@ -13,6 +13,7 @@
 
 ## Breaking changes
 
+- Removed the IREE code from the IDAKLU solver ([#5080](https://github.com/pybamm-team/PyBaMM/pull/5080))
 - Removed support for Python 3.9 ([#5052](https://github.com/pybamm-team/PyBaMM/pull/5052))
 
 # [v25.6.0](https://github.com/pybamm-team/PyBaMM/tree/v25.6.0) - 2025-05-27

src/pybamm/__init__.py

@@ -1,8 +1,5 @@
 from pybamm.version import __version__
 
-# Demote expressions to 32-bit floats/ints - option used for IDAKLU-MLIR compilation
-demote_expressions_to_32bit = False
-
 # Utility classes and methods
 from .util import root_dir
 from .util import Timer, TimerTime, FuzzyDict
@@ -174,7 +171,7 @@
 from .solvers.jax_bdf_solver import jax_bdf_integrate
 
 from .solvers.idaklu_jax import IDAKLUJax
-from .solvers.idaklu_solver import IDAKLUSolver, has_iree
+from .solvers.idaklu_solver import IDAKLUSolver
 
 # Experiments
 from .experiment.experiment import Experiment

src/pybamm/expression_tree/operations/evaluate_python.py

@@ -603,54 +603,9 @@ def __init__(self, symbol: pybamm.Symbol):
             static_argnums=self._static_argnums,
         )
 
-    def _demote_constants(self):
-        """Demote 64-bit constants (f64, i64) to 32-bit (f32, i32)"""
-        if not pybamm.demote_expressions_to_32bit:
-            return  # pragma: no cover
-        self._constants = EvaluatorJax._demote_64_to_32(self._constants)
-
-    @classmethod
-    def _demote_64_to_32(cls, c):
-        """Demote 64-bit operations (f64, i64) to 32-bit (f32, i32)"""
-
-        if not pybamm.demote_expressions_to_32bit:
-            return c
-        if isinstance(c, float):
-            c = jax.numpy.float32(c)
-        if isinstance(c, int):
-            c = jax.numpy.int32(c)
-        if isinstance(c, np.int64):
-            c = c.astype(jax.numpy.int32)
-        if isinstance(c, np.ndarray):
-            if c.dtype == np.float64:
-                c = c.astype(jax.numpy.float32)
-            if c.dtype == np.int64:
-                c = c.astype(jax.numpy.int32)
-        if isinstance(c, jax.numpy.ndarray):
-            if c.dtype == jax.numpy.float64:
-                c = c.astype(jax.numpy.float32)
-            if c.dtype == jax.numpy.int64:
-                c = c.astype(jax.numpy.int32)
-        if isinstance(
-            c, pybamm.expression_tree.operations.evaluate_python.JaxCooMatrix
-        ):
-            if c.data.dtype == np.float64:
-                c.data = c.data.astype(jax.numpy.float32)
-            if c.row.dtype == np.int64:
-                c.row = c.row.astype(jax.numpy.int32)
-            if c.col.dtype == np.int64:
-                c.col = c.col.astype(jax.numpy.int32)
-        if isinstance(c, dict):
-            c = {key: EvaluatorJax._demote_64_to_32(value) for key, value in c.items()}
-        if isinstance(c, tuple):
-            c = tuple(EvaluatorJax._demote_64_to_32(value) for value in c)
-        if isinstance(c, list):
-            c = [EvaluatorJax._demote_64_to_32(value) for value in c]
-        return c
-
     @property
     def _constants(self):
-        return tuple(map(EvaluatorJax._demote_64_to_32, self.__constants))
+        return self.__constants
 
     @_constants.setter
     def _constants(self, value):