StepCache

class cubie.integrators.algorithms.base_algorithm_step.StepCache(step: Callable, nonlinear_solver: Callable | None = None)[source]

Bases: CUDADispatcherCache

Container for compiled device helpers used by an algorithm step.

Parameters:

  • step (Callable) – Device function that advances the integration state.

  • nonlinear_solver (Callable | None) – Optional device function used by implicit methods to perform nonlinear solves.

nonlinear_solver: Callable | None
step: Callable