ODECache

class cubie.odesystems.ODECache(dxdt: Callable | None, linear_operator: Callable | int | None = -1, linear_operator_cached: Callable | int | None = -1, neumann_preconditioner: Callable | int | None = -1, neumann_preconditioner_cached: Callable | int | None = -1, stage_residual: Callable | int | None = -1, n_stage_residual: Callable | int | None = -1, n_stage_linear_operator: Callable | int | None = -1, n_stage_neumann_preconditioner: Callable | int | None = -1, observables: Callable | int | None = -1, prepare_jac: Callable | int | None = -1, calculate_cached_jvp: Callable | int | None = -1, time_derivative_rhs: Callable | int | None = -1, cached_aux_count: int | None = -1)

Bases: CUDADispatcherCache

Cache compiled CUDA device and support functions for an ODE system.

Attributes default to -1 when the corresponding function is not built.

cached_aux_count: int | None

calculate_cached_jvp: Callable | int | None

dxdt: Callable | None

linear_operator: Callable | int | None

linear_operator_cached: Callable | int | None

n_stage_linear_operator: Callable | int | None

n_stage_neumann_preconditioner: Callable | int | None

n_stage_residual: Callable | int | None

neumann_preconditioner: Callable | int | None

neumann_preconditioner_cached: Callable | int | None

observables: Callable | int | None

prepare_jac: Callable | int | None

stage_residual: Callable | int | None

time_derivative_rhs: Callable | int | None