GEOS
FluidUpdateKernel.hpp
Go to the documentation of this file.
1 /*
2  * ------------------------------------------------------------------------------------------------------------
3  * SPDX-License-Identifier: LGPL-2.1-only
4  *
5  * Copyright (c) 2016-2024 Lawrence Livermore National Security LLC
6  * Copyright (c) 2018-2024 Total, S.A
7  * Copyright (c) 2018-2024 The Board of Trustees of the Leland Stanford Junior University
8  * Copyright (c) 2023-2024 Chevron
9  * Copyright (c) 2019- GEOS/GEOSX Contributors
10  * All rights reserved
11  *
12  * See top level LICENSE, COPYRIGHT, CONTRIBUTORS, NOTICE, and ACKNOWLEDGEMENTS files for details.
13  * ------------------------------------------------------------------------------------------------------------
14  */
15 
20 #ifndef GEOS_PHYSICSSOLVERS_FLUIDFLOW_COMPOSITIONAL_FLUIDUPDATEKERNEL_HPP
21 #define GEOS_PHYSICSSOLVERS_FLUIDFLOW_COMPOSITIONAL_FLUIDUPDATEKERNEL_HPP
22 
23 #include "common/DataTypes.hpp"
24 #include "common/GEOS_RAJA_Interface.hpp"
25 
26 namespace geos
27 {
28 
29 namespace thermalCompositionalMultiphaseBaseKernels
30 {
31 
32 /******************************** FluidUpdateKernel ********************************/
33 
35 {
36  template< typename POLICY, typename FLUID_WRAPPER >
37  static void
38  launch( localIndex const size,
39  FLUID_WRAPPER const & fluidWrapper,
40  arrayView1d< real64 const > const & pres,
41  arrayView1d< real64 const > const & temp,
43  {
44  forAll< POLICY >( size, [=] GEOS_HOST_DEVICE ( localIndex const k )
45  {
46  for( localIndex q = 0; q < fluidWrapper.numGauss(); ++q )
47  {
48  fluidWrapper.update( k, q, pres[k], temp[k], compFrac[k] );
49  }
50  } );
51  }
52 
53  template< typename POLICY, typename FLUID_WRAPPER >
54  static void
55  launch( SortedArrayView< localIndex const > const & targetSet,
56  FLUID_WRAPPER const & fluidWrapper,
57  arrayView1d< real64 const > const & pres,
58  arrayView1d< real64 const > const & temp,
60  {
61  forAll< POLICY >( targetSet.size(), [=] GEOS_HOST_DEVICE ( localIndex const a )
62  {
63  localIndex const k = targetSet[a];
64  for( localIndex q = 0; q < fluidWrapper.numGauss(); ++q )
65  {
66  fluidWrapper.update( k, q, pres[k], temp[k], compFrac[k] );
67  }
68  } );
69  }
70 };
71 
72 } // namespace thermalCompositionalMultiphaseBaseKernels
73 
74 } // namespace geos
75 
76 
77 #endif //GEOS_PHYSICSSOLVERS_FLUIDFLOW_THERMALCOMPOSITIONALMULTIPHASEBASEKERNELS_HPP
#define GEOS_HOST_DEVICE
Marks a host-device function.
Definition: GeosxMacros.hpp:49
ArrayView< T, 1 > arrayView1d
Alias for 1D array view.
Definition: DataTypes.hpp:180
GEOS_LOCALINDEX_TYPE localIndex
Local index type (for indexing objects within an MPI partition).
Definition: DataTypes.hpp:85
LvArray::SortedArrayView< T, localIndex, LvArray::ChaiBuffer > SortedArrayView
A sorted array view of local indices.
Definition: DataTypes.hpp:271
ArrayView< T, 2, USD > arrayView2d
Alias for 2D array view.
Definition: DataTypes.hpp:196