ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update Kokkos library in LAMMPS to v3.6.0

Browse Source
This commit is contained in:
Stan Gerald Moore
2022-05-05 11:44:47 -06:00
parent bd4bbbddbe
commit b79c0bc7b4
380 changed files with 41928 additions and 8786 deletions
Download Patch File Download Diff File Expand all files Collapse all files

92
lib/kokkos/core/src/Kokkos_Macros.hpp
View File

@ -80,7 +80,7 @@
* KOKKOS_COMPILER_PGI
* KOKKOS_COMPILER_MSVC
*
* Macros for which compiler extension to use for atomics on intrinsice types
* Macros for which compiler extension to use for atomics on intrinsic types
*
* KOKKOS_ENABLE_CUDA_ATOMICS
* KOKKOS_ENABLE_GNU_ATOMICS
@ -187,6 +187,12 @@
#endif
#endif
#if defined(__NVCOMPILER)
#define KOKKOS_COMPILER_NVHPC \
__NVCOMPILER_MAJOR__ * 100 + __NVCOMPILER_MINOR__ * 10 + \
__NVCOMPILER_PATCHLEVEL__
#endif
#if defined(_MSC_VER) && !defined(KOKKOS_COMPILER_INTEL)
#define KOKKOS_COMPILER_MSVC _MSC_VER
#endif
@ -198,6 +204,16 @@
// of the supported OpenMP API version.
#endif // #if defined( _OPENMP )
#if defined(KOKKOS_ENABLE_CXX17)
#define KOKKOS_IMPL_FALLTHROUGH [[fallthrough]];
#elif defined(KOKKOS_COMPILER_GNU) && (KOKKOS_COMPILER_GNU >= 710)
#define KOKKOS_IMPL_FALLTHROUGH [[gnu::fallthrough]];
#elif defined(KOKKOS_COMPILER_CLANG)
#define KOKKOS_IMPL_FALLTHROUGH [[clang::fallthrough]];
#else
#define KOKKOS_IMPL_FALLTHROUGH
#endif
//----------------------------------------------------------------------------
// Intel compiler macros
@ -471,11 +487,6 @@
#define KOKKOS_ENABLE_DEFAULT_DEVICE_TYPE_CUDA
#elif defined(KOKKOS_ENABLE_HIP)
#define KOKKOS_ENABLE_DEFAULT_DEVICE_TYPE_HIP
#if defined(__HIP__)
// mark that HIP-clang can use __host__ and __device__
// as valid overload criteria
#define KOKKOS_IMPL_ENABLE_OVERLOAD_HOST_DEVICE
#endif
#elif defined(KOKKOS_ENABLE_SYCL)
#define KOKKOS_ENABLE_DEFAULT_DEVICE_TYPE_SYCL
#elif defined(KOKKOS_ENABLE_OPENMPTARGET)
@ -506,6 +517,69 @@
//----------------------------------------------------------------------------
// Remove surrounding parentheses if present
#define KOKKOS_IMPL_STRIP_PARENS(X) KOKKOS_IMPL_ESC(KOKKOS_IMPL_ISH X)
#define KOKKOS_IMPL_ISH(...) KOKKOS_IMPL_ISH __VA_ARGS__
#define KOKKOS_IMPL_ESC(...) KOKKOS_IMPL_ESC_(__VA_ARGS__)
#define KOKKOS_IMPL_ESC_(...) KOKKOS_IMPL_VAN_##__VA_ARGS__
#define KOKKOS_IMPL_VAN_KOKKOS_IMPL_ISH
#if defined(KOKKOS_ENABLE_CUDA) && defined(KOKKOS_COMPILER_NVHPC)
#include <nv/target>
#define KOKKOS_IF_ON_DEVICE(CODE) NV_IF_TARGET(NV_IS_DEVICE, CODE)
#define KOKKOS_IF_ON_HOST(CODE) NV_IF_TARGET(NV_IS_HOST, CODE)
#endif
#ifdef KOKKOS_ENABLE_OPENMPTARGET
#ifdef KOKKOS_COMPILER_NVHPC
#define KOKKOS_IF_ON_DEVICE(CODE) \
if (__builtin_is_device_code()) { \
KOKKOS_IMPL_STRIP_PARENS(CODE) \
}
#define KOKKOS_IF_ON_HOST(CODE) \
if (!__builtin_is_device_code()) { \
KOKKOS_IMPL_STRIP_PARENS(CODE) \
}
#else
// Base function.
static constexpr bool kokkos_omp_on_host() { return true; }
#pragma omp begin declare variant match(device = {kind(host)})
static constexpr bool kokkos_omp_on_host() { return true; }
#pragma omp end declare variant
#pragma omp begin declare variant match(device = {kind(nohost)})
static constexpr bool kokkos_omp_on_host() { return false; }
#pragma omp end declare variant
#define KOKKOS_IF_ON_DEVICE(CODE) \
if constexpr (!kokkos_omp_on_host()) { \
KOKKOS_IMPL_STRIP_PARENS(CODE) \
}
#define KOKKOS_IF_ON_HOST(CODE) \
if constexpr (kokkos_omp_on_host()) { \
KOKKOS_IMPL_STRIP_PARENS(CODE) \
}
#endif
#endif
#if !defined(KOKKOS_IF_ON_HOST) && !defined(KOKKOS_IF_ON_DEVICE)
#if defined(__CUDA_ARCH__) || defined(__HIP_DEVICE_COMPILE__) || \
defined(__SYCL_DEVICE_ONLY__)
#define KOKKOS_IF_ON_DEVICE(CODE) \
{ KOKKOS_IMPL_STRIP_PARENS(CODE) }
#define KOKKOS_IF_ON_HOST(CODE) \
{}
#else
#define KOKKOS_IF_ON_DEVICE(CODE) \
{}
#define KOKKOS_IF_ON_HOST(CODE) \
{ KOKKOS_IMPL_STRIP_PARENS(CODE) }
#endif
#endif
//----------------------------------------------------------------------------
#if (defined(_POSIX_C_SOURCE) && _POSIX_C_SOURCE >= 200112L) || \
(defined(_XOPEN_SOURCE) && _XOPEN_SOURCE >= 600)
#if defined(KOKKOS_ENABLE_PERFORMANCE_POSIX_MEMALIGN)
@ -514,8 +588,8 @@
#endif
//----------------------------------------------------------------------------
// If compiling with CUDA, we must use relocateable device code
// to enable the task policy.
// If compiling with CUDA, we must use relocatable device code to enable the
// task policy.
#if defined(KOKKOS_ENABLE_CUDA)
#if defined(KOKKOS_ENABLE_CUDA_RELOCATABLE_DEVICE_CODE)
@ -539,7 +613,7 @@
// intel error #2651: attribute does not apply to any entity
// using <deprecated_type> KOKKOS_DEPRECATED = ...
#if defined(KOKKOS_ENABLE_DEPRECATION_WARNINGS) && !defined(__NVCC__) && \
(KOKKOS_COMPILER_INTEL > 1900)
(!defined(KOKKOS_COMPILER_INTEL) || KOKKOS_COMPILER_INTEL > 1900)
#define KOKKOS_DEPRECATED [[deprecated]]
#define KOKKOS_DEPRECATED_WITH_COMMENT(comment) [[deprecated(comment)]]
#else