spectral_galerkin_heat.backends.cupy_backend

GPU math backend: CuPy arrays with Numba CUDA kernels.

This module imports cupy, so it is imported lazily (only from spectral_galerkin_heat.backends.get_backend()) to keep CuPy an optional dependency.

Classes

CupyBackend

GPU backend: CuPy arrays and Numba CUDA kernels.

Module Contents

class spectral_galerkin_heat.backends.cupy_backend.CupyBackend

Bases: spectral_galerkin_heat.backends.base.MathBackend

GPU backend: CuPy arrays and Numba CUDA kernels.