spectral_galerkin_heat.physics.spectral_gpu_kernels

GPU-based spectral method kernels for the heat equation (requires CuPy).

Public functions in this module mirror CPU kernels and are called by SpectralSolver (CupyBackend).

Classes

SpectralSolverState

GPU spectral state: spectral_state.SpectralSolverState bound to CuPy.

Functions

update_modes_etd1(aK, KK, Cp_broadcast, B_scaled, ...)	Wrapper for ETD1 kernel.
add_source_term_modes(a_temp, KK, Q_modes)	Wrapper for Source Term Accumulation.
add_bottom_surface_source(a_temp, KK, ...)	Wrapper for bottom surface source accumulation (GPU).

Module Contents

class spectral_galerkin_heat.physics.spectral_gpu_kernels.SpectralSolverState(phys, geom, num, fine)

Bases: spectral_galerkin_heat.physics.spectral_state.SpectralSolverState

GPU spectral state: spectral_state.SpectralSolverState bound to CuPy.

spectral_galerkin_heat.physics.spectral_gpu_kernels.update_modes_etd1(aK, KK, Cp_broadcast, B_scaled, a_temp_out)

Wrapper for ETD1 kernel.

spectral_galerkin_heat.physics.spectral_gpu_kernels.add_source_term_modes(a_temp, KK, Q_modes)

Wrapper for Source Term Accumulation.

spectral_galerkin_heat.physics.spectral_gpu_kernels.add_bottom_surface_source(a_temp, KK, Cp_broadcast_bottom, B_scaled)

Wrapper for bottom surface source accumulation (GPU).