spectral_galerkin_heat.physics.spectral_gpu_kernels =================================================== .. py:module:: spectral_galerkin_heat.physics.spectral_gpu_kernels .. autoapi-nested-parse:: 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 ------- .. autoapisummary:: spectral_galerkin_heat.physics.spectral_gpu_kernels.SpectralSolverState Functions --------- .. autoapisummary:: spectral_galerkin_heat.physics.spectral_gpu_kernels.update_modes_etd1 spectral_galerkin_heat.physics.spectral_gpu_kernels.add_source_term_modes spectral_galerkin_heat.physics.spectral_gpu_kernels.add_bottom_surface_source Module Contents --------------- .. py:class:: SpectralSolverState(phys, geom, num, fine) Bases: :py:obj:`spectral_galerkin_heat.physics.spectral_state.SpectralSolverState` GPU spectral state: :class:`spectral_state.SpectralSolverState` bound to CuPy. .. py:function:: update_modes_etd1(aK, KK, Cp_broadcast, B_scaled, a_temp_out) Wrapper for ETD1 kernel. .. py:function:: add_source_term_modes(a_temp, KK, Q_modes) Wrapper for Source Term Accumulation. .. py:function:: add_bottom_surface_source(a_temp, KK, Cp_broadcast_bottom, B_scaled) Wrapper for bottom surface source accumulation (GPU).