GlobalAdaptiveTimeStepWithEnclaveTasking.py

Go to the documentation of this file.

# This file is part of the ExaHyPE2 project. For conditions of distribution and
# use, please see the copyright notice at www.peano-framework.org
from exahype2.solvers.PDETerms                         import PDETerms
from exahype2.solvers.rkdg.SeparateSweepsWithEnclaveTasking import SeparateSweepsWithEnclaveTasking
 
from exahype2.solvers.rkdg.kernels import create_abstract_solver_declarations
from exahype2.solvers.rkdg.kernels import create_abstract_solver_definitions
from exahype2.solvers.rkdg.kernels import create_solver_declarations
from exahype2.solvers.rkdg.kernels import create_solver_definitions
 
from exahype2.solvers.rkdg.kernels import SolverVariant
 
from exahype2.solvers.rkdg.kernels         import create_volumetric_solver_call
from exahype2.solvers.rkdg.kernels         import create_add_solver_contributions_call
from exahype2.solvers.rkdg.kernels         import create_multiply_with_inverted_mass_matrix_call
from exahype2.solvers.rkdg.rusanov.kernels import create_Riemann_solver_call
 
from exahype2.solvers.rkdg.AdaptiveTimeSteppingCodeSnippets import AdaptiveTimeSteppingCodeSnippets
 
from exahype2.solvers.rkdg.actionsets.ProjectLinearCombinationOfEstimatesOntoFaces import FaceProjections
 
 
class GlobalAdaptiveTimeStepWithEnclaveTasking(SeparateSweepsWithEnclaveTasking):
  """!
  
  RKDG solver with global adaptive time step 
  
  Please consult GlobalAdaptiveTimeStep for more documentation. This version is
  basically the same with the "only" exception being that we employ enclave
  tasking here.
  
  """  
  def __init__(self,
    name,
    rk_order,
    polynomials,
    unknowns,
    auxiliary_variables,
    min_cell_h,
    max_cell_h,
    time_step_relaxation,
    flux=PDETerms.User_Defined_Implementation,
    eigenvalues=PDETerms.User_Defined_Implementation,
    ncp=PDETerms.None_Implementation,
    point_source=PDETerms.None_Implementation,
    boundary_conditions=PDETerms.User_Defined_Implementation,
    refinement_criterion=PDETerms.Empty_Implementation,
    initial_conditions=PDETerms.User_Defined_Implementation,
    source_term=PDETerms.None_Implementation,
    plot_grid_properties=False,
    pde_terms_without_state=False,
  ):
    """!
        
    Construct RKDG solver with adaptive global time step and enclave tasking
        
    """
    super(GlobalAdaptiveTimeStepWithEnclaveTasking,self).__init__(name,
                                                rk_order,
                                                polynomials,
                                                FaceProjections.Solution,
                                                unknowns,
                                                auxiliary_variables,
                                                min_cell_h,
                                                max_cell_h,
                                                plot_grid_properties,
                                                pde_terms_without_state)
 
    self._time_step_relaxation                                = time_step_relaxation
    self._compute_eigenvalue_compute_eigenvalue                                  = True
 
    self._kernel_namespace_kernel_namespace                                    = "rusanov"
    self._volumetric_compute_kernel_call_volumetric_compute_kernel_call                      = create_volumetric_solver_call(self._basis, SolverVariant.WithVirtualFunctions)
    self._volumetric_compute_kernel_call_stateless_volumetric_compute_kernel_call_stateless            = create_volumetric_solver_call(self._basis, SolverVariant.Stateless)
    self._Riemann_compute_kernel_call_Riemann_compute_kernel_call                         = create_Riemann_solver_call(self._basis, FaceProjections.Solution)
    # TODO: self._Riemann_compute_kernel_call_stateless       =
    self._fused_volumetric_compute_kernel_call_stateless_cpu_fused_volumetric_compute_kernel_call_stateless_cpu  = create_volumetric_solver_call(self._basis, SolverVariant.Stateless)
    self._fused_volumetric_compute_kernel_call_stateless_gpu_fused_volumetric_compute_kernel_call_stateless_gpu  = create_volumetric_solver_call(self._basis, SolverVariant.Accelerator)
    # TODO: self._fused_Riemann_compute_kernel_call_cpu       =
    # TODO: self._fused_Riemann_compute_kernel_call_gpu       =
    self._add_solver_contributions_call_add_solver_contributions_call                       = create_add_solver_contributions_call(self._basis)
    self._multiply_with_inverted_mass_matrix_call_multiply_with_inverted_mass_matrix_call             = create_multiply_with_inverted_mass_matrix_call(self._basis)
 
    self.set_implementationset_implementation(boundary_conditions=boundary_conditions,
                            refinement_criterion=refinement_criterion,
                            initial_conditions=initial_conditions,
                            additional_action_set_includes="",
                            additional_user_includes="",
                            flux=flux,
                            eigenvalues=eigenvalues,
                            ncp=ncp,
                            source_term=source_term,
                            point_source=point_source
                            )
 
 
  def set_implementation(self,
    flux=None,
    ncp=None,
    eigenvalues=None,
    boundary_conditions=None,
    refinement_criterion=None,
    initial_conditions=None,
    source_term=None,
    point_source=None,
    additional_action_set_includes = "",
    additional_user_includes       = ""
  ):
    """
    If you pass in User_Defined, then the generator will create C++ stubs
    that you have to befill manually. If you pass in None_Implementation, it
    will create nop, i.e., no implementation or defaults. Any other string
    is copied 1:1 into the implementation. If you pass in None, then the
    set value so far won't be overwritten.
 
    Please note that not all options are supported by all solvers.
 
    This routine should be the very last invoked by the constructor.
    """
    super(GlobalAdaptiveTimeStepWithEnclaveTasking,self).set_implementation(boundary_conditions=boundary_conditions,
                                                          refinement_criterion=refinement_criterion,
                                                          initial_conditions=initial_conditions,
                                                          additional_action_set_includes=additional_action_set_includes,
                                                          additional_user_includes=additional_user_includes,
                                                          flux=flux,
                                                          ncp=ncp,
                                                          eigenvalues=eigenvalues,
                                                          source_term=source_term,
                                                          point_source=point_source
                                                          )
 
    solver_code_snippets = AdaptiveTimeSteppingCodeSnippets(self._time_step_relaxation)
 
    self._abstract_solver_user_declarations_abstract_solver_user_declarations  = create_abstract_solver_declarations(self._flux_implementation, self._ncp_implementation, self._eigenvalues_implementation, self._source_term_implementation, self._point_sources_implementation, pde_terms_without_state=self._pde_terms_without_state_pde_terms_without_state)
    self._abstract_solver_user_declarations_abstract_solver_user_declarations += solver_code_snippets.create_abstract_solver_user_declarations()
    self._abstract_solver_user_definitions_abstract_solver_user_definitions   = create_abstract_solver_definitions( self._flux_implementation, self._ncp_implementation, self._eigenvalues_implementation, self._source_term_implementation, self._point_sources_implementation, pde_terms_without_state=self._pde_terms_without_state_pde_terms_without_state)
    self._abstract_solver_user_definitions_abstract_solver_user_definitions  += solver_code_snippets.create_abstract_solver_user_definitions()
 
    self._solver_user_declarations_solver_user_declarations           = create_solver_declarations(self._flux_implementation, self._ncp_implementation, self._eigenvalues_implementation, self._source_term_implementation, self._point_sources_implementation, pde_terms_without_state=self._pde_terms_without_state_pde_terms_without_state)
    self._solver_user_definitions_solver_user_definitions            = create_solver_definitions( self._flux_implementation, self._ncp_implementation, self._eigenvalues_implementation, self._source_term_implementation, self._point_sources_implementation, pde_terms_without_state=self._pde_terms_without_state_pde_terms_without_state)
 
    self._compute_time_step_size_compute_time_step_size             = solver_code_snippets.create_compute_time_step_size()
    self._compute_new_time_step_size_compute_new_time_step_size         = solver_code_snippets.create_compute_new_time_step_size()
 
    self._start_time_step_implementation_start_time_step_implementation     = solver_code_snippets.create_start_time_step_implementation()
    self._finish_time_step_implementation_finish_time_step_implementation    = solver_code_snippets.create_finish_time_step_implementation()
    self._constructor_implementation_constructor_implementation         = solver_code_snippets.create_abstract_solver_constructor_statements()
 
 
  @property
  def user_action_set_includes(self):
    return super(GlobalAdaptiveTimeStepWithEnclaveTasking, self).user_action_set_includes + """
#include "exahype2/dg/rusanov/Rusanov.h"
"""