DataLayouts.hpp

Go to the documentation of this file.

/*
 * ------------------------------------------------------------------------------------------------------------
 * SPDX-License-Identifier: LGPL-2.1-only
 *
 * Copyright (c) 2018-2020 Lawrence Livermore National Security LLC
 * Copyright (c) 2018-2020 The Board of Trustees of the Leland Stanford Junior University
 * Copyright (c) 2018-2020 Total, S.A
 * Copyright (c) 2019-     GEOSX Contributors
 * All rights reserved
 *
 * See top level LICENSE, COPYRIGHT, CONTRIBUTORS, NOTICE, and ACKNOWLEDGEMENTS files for details.
 * ------------------------------------------------------------------------------------------------------------
 */

#ifndef GEOSX_COMMON_DATALAYOUTS_HPP_
#define GEOSX_COMMON_DATALAYOUTS_HPP_

#include "RAJA/RAJA.hpp"
#include "LvArray/src/Array.hpp"

namespace geosx
{

namespace nodes
{

#if defined( GEOSX_USE_CUDA )

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 cells
{

#if defined( GEOSX_USE_CUDA )

using NODE_MAP_PERMUTATION = RAJA::PERM_JI;

#else

using NODE_MAP_PERMUTATION = RAJA::PERM_IJ;

#endif

static constexpr int NODE_MAP_USD = LvArray::typeManipulation::getStrideOneDimension( NODE_MAP_PERMUTATION {} );

} // namespace cells

namespace solid
{

#if defined( GEOSX_USE_CUDA )

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 geosx

#endif // GEOSX_COMMON_DATALAYOUTS_HPP_