DataLayouts.hpp

Go to the documentation of this file.

/*
 * ------------------------------------------------------------------------------------------------------------
 * SPDX-License-Identifier: LGPL-2.1-only
 *
 * Copyright (c) 2016-2024 Lawrence Livermore National Security LLC
 * Copyright (c) 2018-2024 TotalEnergies
 * Copyright (c) 2018-2024 The Board of Trustees of the Leland Stanford Junior University
 * Copyright (c) 2023-2024 Chevron
 * Copyright (c) 2019-     GEOS/GEOSX Contributors
 * All rights reserved
 *
 * See top level LICENSE, COPYRIGHT, CONTRIBUTORS, NOTICE, and ACKNOWLEDGEMENTS files for details.
 * ------------------------------------------------------------------------------------------------------------
 */
 
#ifndef GEOS_COMMON_DATALAYOUTS_HPP_
#define GEOS_COMMON_DATALAYOUTS_HPP_
 
#include "common/GeosxConfig.hpp"
 
#include "LvArray/src/Array.hpp"
#include "RAJA/RAJA.hpp"
 
namespace geos
{
 
template< int NDIM >
using defaultLayout = camp::make_idx_seq_t< NDIM >;
 
template< typename PERM >
static constexpr int getUSD = LvArray::typeManipulation::getStrideOneDimension( PERM {} );
 
namespace nodes
{
 
#if defined( GEOS_USE_DEVICE )
 
using REFERENCE_POSITION_PERM = RAJA::PERM_JI;
 
using TOTAL_DISPLACEMENT_PERM = RAJA::PERM_JI;
 
using INCR_DISPLACEMENT_PERM = RAJA::PERM_JI;
 
using VELOCITY_PERM = RAJA::PERM_JI;
 
using ACCELERATION_PERM = RAJA::PERM_JI;
 
#else
 
using REFERENCE_POSITION_PERM = RAJA::PERM_IJ;
 
using TOTAL_DISPLACEMENT_PERM = RAJA::PERM_IJ;
 
using INCR_DISPLACEMENT_PERM = RAJA::PERM_IJ;
 
using VELOCITY_PERM = RAJA::PERM_IJ;
 
using ACCELERATION_PERM = RAJA::PERM_IJ;
 
#endif
 
static constexpr int REFERENCE_POSITION_USD = LvArray::typeManipulation::getStrideOneDimension( REFERENCE_POSITION_PERM {} );
 
static constexpr int TOTAL_DISPLACEMENT_USD = LvArray::typeManipulation::getStrideOneDimension( TOTAL_DISPLACEMENT_PERM {} );
 
static constexpr int INCR_DISPLACEMENT_USD = LvArray::typeManipulation::getStrideOneDimension( INCR_DISPLACEMENT_PERM {} );
 
static constexpr int VELOCITY_USD = LvArray::typeManipulation::getStrideOneDimension( VELOCITY_PERM {} );
 
static constexpr int ACCELERATION_USD = LvArray::typeManipulation::getStrideOneDimension( ACCELERATION_PERM {} );
 
} // namespace nodes
 
namespace particles
{
 
#if defined( GEOS_USE_CUDA )
 
using REFERENCE_POSITION_PERM = RAJA::PERM_JI;
 
using TOTAL_DISPLACEMENT_PERM = RAJA::PERM_JI;
 
using VELOCITY_PERM = RAJA::PERM_JI;
 
using ACCELERATION_PERM = RAJA::PERM_JI;
 
#else
 
using REFERENCE_POSITION_PERM = RAJA::PERM_IJ;
 
using TOTAL_DISPLACEMENT_PERM = RAJA::PERM_IJ;
 
using VELOCITY_PERM = RAJA::PERM_IJ;
 
using ACCELERATION_PERM = RAJA::PERM_IJ;
 
#endif
 
static constexpr int REFERENCE_POSITION_USD = LvArray::typeManipulation::getStrideOneDimension( REFERENCE_POSITION_PERM {} );
 
static constexpr int TOTAL_DISPLACEMENT_USD = LvArray::typeManipulation::getStrideOneDimension( TOTAL_DISPLACEMENT_PERM {} );
 
static constexpr int VELOCITY_USD = LvArray::typeManipulation::getStrideOneDimension( VELOCITY_PERM {} );
 
static constexpr int ACCELERATION_USD = LvArray::typeManipulation::getStrideOneDimension( ACCELERATION_PERM {} );
 
} // namespace particles
 
namespace cells
{
 
#if defined( GEOS_USE_DEVICE )
 
using NODE_MAP_PERMUTATION = RAJA::PERM_JI;
 
using STRAIN_PERM = RAJA::PERM_JI;
 
#else
 
using NODE_MAP_PERMUTATION = RAJA::PERM_IJ;
 
using STRAIN_PERM = RAJA::PERM_IJ;
 
#endif
 
static constexpr int NODE_MAP_USD = LvArray::typeManipulation::getStrideOneDimension( NODE_MAP_PERMUTATION {} );
 
static constexpr int STRAIN_USD = LvArray::typeManipulation::getStrideOneDimension( STRAIN_PERM {} );
 
} // namespace cells
 
namespace solid
{
 
#if defined( GEOS_USE_DEVICE )
 
using STRESS_PERMUTATION = RAJA::PERM_KJI;
 
using STIFFNESS_PERMUTATION = RAJA::PERM_KJI;
 
#else
 
using STRESS_PERMUTATION = RAJA::PERM_IJK;
 
using STIFFNESS_PERMUTATION = RAJA::PERM_IJK;
 
#endif
 
static constexpr int STRESS_USD = LvArray::typeManipulation::getStrideOneDimension( STRESS_PERMUTATION {} );
 
static constexpr int STIFFNESS_USD = LvArray::typeManipulation::getStrideOneDimension( STIFFNESS_PERMUTATION {} );
 
} // namespace solid
 
namespace compflow
{
 
#if defined( GEOS_USE_DEVICE )
 
using LAYOUT_COMP = RAJA::PERM_JI;
 
using LAYOUT_COMP_DC = RAJA::PERM_JKI;
 
using LAYOUT_PHASE = RAJA::PERM_JI;
 
using LAYOUT_PHASE_DC = RAJA::PERM_JKI;
 
using LAYOUT_PHASE_COMP = RAJA::PERM_JKI;
 
using LAYOUT_FLUID_DC = RAJA::PERM_JI;
 
using LAYOUT_OBL_OPERATOR_VALUES = RAJA::PERM_JI;
 
using LAYOUT_OBL_OPERATOR_DERIVATIVES = RAJA::PERM_JKI;
 
#else
 
using LAYOUT_COMP = RAJA::PERM_IJ;
 
using LAYOUT_COMP_DC = RAJA::PERM_IJK;
 
using LAYOUT_PHASE = RAJA::PERM_IJ;
 
using LAYOUT_PHASE_DC = RAJA::PERM_IJK;
 
using LAYOUT_PHASE_COMP = RAJA::PERM_IJK;
 
using LAYOUT_FLUID_DC = RAJA::PERM_IJ;
 
using LAYOUT_OBL_OPERATOR_VALUES = RAJA::PERM_IJ;
 
using LAYOUT_OBL_OPERATOR_DERIVATIVES = RAJA::PERM_IJK;
 
#endif
 
static constexpr int USD_COMP = LvArray::typeManipulation::getStrideOneDimension( LAYOUT_COMP{} );
 
static constexpr int USD_COMP_DC = LvArray::typeManipulation::getStrideOneDimension( LAYOUT_COMP_DC{} );
 
static constexpr int USD_PHASE = LvArray::typeManipulation::getStrideOneDimension( LAYOUT_PHASE{} );
 
static constexpr int USD_PHASE_DC = LvArray::typeManipulation::getStrideOneDimension( LAYOUT_PHASE_DC{} );
 
static constexpr int USD_PHASE_COMP = LvArray::typeManipulation::getStrideOneDimension( LAYOUT_PHASE_COMP{} );
 
static constexpr int USD_FLUID_DC = LvArray::typeManipulation::getStrideOneDimension( LAYOUT_FLUID_DC{} );
 
static constexpr int USD_OBL_VAL = LvArray::typeManipulation::getStrideOneDimension( LAYOUT_OBL_OPERATOR_VALUES{} );
 
static constexpr int USD_OBL_DER = LvArray::typeManipulation::getStrideOneDimension( LAYOUT_OBL_OPERATOR_DERIVATIVES{} );
 
} // namespace compflow
 
} // namespace geos
 
#endif // GEOS_COMMON_DATALAYOUTS_HPP_