compute_hess_kernel.rb

module BOAST
  def BOAST::compute_hess_kernel(ref = true, n_gll3 = 125)
    push_env( :array_start => 0 )
    kernel = CKernel::new
    function_name = "compute_hess_kernel"
    ibool   = Int( "ibool",   :dir => :in, :dim => [Dim()] )
    accel   = Real("accel",   :dir => :in, :dim => [Dim(3), Dim()] )
    b_accel = Real("b_accel", :dir => :in, :dim => [Dim(3), Dim()] )
    hess_kl = Real("hess_kl",  :dir => :inout,:dim => [Dim()] )
    deltat  = Real("deltat",  :dir => :in)
    nspec_ab= Int( "NSPEC_AB",   :dir => :in)
    use_source_receiver_hess = Int("USE_SOURCE_RECEIVER_HESSIAN", :dir => :in)

    ngll3 = Int("NGLL3", :const => n_gll3)

    p = Procedure(function_name, [ibool, accel, b_accel, hess_kl, deltat, nspec_ab, use_source_receiver_hess])
    if (get_lang == CUDA and ref) then
      get_output.print File::read("references/#{function_name}.cu")
    elsif(get_lang == CL or get_lang == CUDA or get_lang == HIP) then
      make_specfem3d_header( :ngll3 => n_gll3 )
      open p
      decl ispec = Int("ispec")
      decl ijk_ispec = Int("ijk_ispec")
      decl iglob = Int("iglob")
      comment()

      print ispec === get_group_id(0) + get_group_id(1)*get_num_groups(0)
      comment()

      print If(ispec < nspec_ab) {
        print ijk_ispec === get_local_id(0) + ngll3*ispec
        print iglob === ibool[ijk_ispec] - 1
        comment()

        print If(use_source_receiver_hess => lambda {
          print hess_kl[ijk_ispec] === hess_kl[ijk_ispec] + deltat * ( accel[0, iglob] * b_accel[0, iglob]\
                                                                     + accel[1, iglob] * b_accel[1, iglob]\
                                                                     + accel[2, iglob] * b_accel[2, iglob])
        }, :else => lambda {
          print hess_kl[ijk_ispec] === hess_kl[ijk_ispec] + deltat * ( accel[0, iglob] * accel[0, iglob]\
                                                                     + accel[1, iglob] * accel[1, iglob]\
                                                                     + accel[2, iglob] * accel[2, iglob])
        })
      }
      close p
    else
      raise "Unsupported language!"
    end
    pop_env( :array_start )
    kernel.procedure = p
    return kernel
  end
end