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

Update Kokkos library in LAMMPS to v3.2

Browse Source
This commit is contained in:
Stan Moore
2020-08-25 20:21:48 -06:00
parent 450fd12d31
commit 4d90c2b74b
1410 changed files with 19364 additions and 71953 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
lib/kokkos/core/src/Kokkos_Atomic.hpp
View File

@ -75,6 +75,9 @@
//----------------------------------------------------------------------------
#if defined(_WIN32)
#define KOKKOS_ENABLE_WINDOWS_ATOMICS
#if defined(KOKKOS_ENABLE_CUDA)
#define KOKKOS_ENABLE_CUDA_ATOMICS
#endif
#else
#if defined(KOKKOS_ENABLE_CUDA)
@ -86,7 +89,8 @@
#define KOKKOS_ENABLE_ROCM_ATOMICS
#elif defined(KOKKOS_ACTIVE_EXECUTION_MEMORY_SPACE_HIP_GPU)
#elif defined(KOKKOS_ACTIVE_EXECUTION_MEMORY_SPACE_HIP_GPU) || \
defined(KOKKOS_IMPL_ENABLE_OVERLOAD_HOST_DEVICE)
#define KOKKOS_ENABLE_HIP_ATOMICS
@ -188,7 +192,7 @@ extern KOKKOS_INLINE_FUNCTION void unlock_address_rocm_space(void* ptr);
#ifdef _WIN32
#include "impl/Kokkos_Atomic_Windows.hpp"
#else
#endif
//----------------------------------------------------------------------------
// Atomic Assembly
//
@ -196,6 +200,15 @@ extern KOKKOS_INLINE_FUNCTION void unlock_address_rocm_space(void* ptr);
#include "impl/Kokkos_Atomic_Assembly.hpp"
//----------------------------------------------------------------------------
// Memory fence
//
// All loads and stores from this thread will be globally consistent before
// continuing
//
// void memory_fence() {...};
#include "impl/Kokkos_Memory_Fence.hpp"
//----------------------------------------------------------------------------
// Atomic exchange
//
@ -215,11 +228,8 @@ extern KOKKOS_INLINE_FUNCTION void unlock_address_rocm_space(void* ptr);
#include "impl/Kokkos_Atomic_Compare_Exchange_Strong.hpp"
#endif //_WIN32
#include "impl/Kokkos_Atomic_Generic.hpp"
#ifndef _WIN32
//----------------------------------------------------------------------------
// Atomic fetch and add
//
@ -285,16 +295,6 @@ extern KOKKOS_INLINE_FUNCTION void unlock_address_rocm_space(void* ptr);
// { T tmp = *dest ; *dest = max(*dest, val); return tmp ; }
#include "impl/Kokkos_Atomic_MinMax.hpp"
#endif /*Not _WIN32*/
//----------------------------------------------------------------------------
// Memory fence
//
// All loads and stores from this thread will be globally consistent before
// continuing
//
// void memory_fence() {...};
#include "impl/Kokkos_Memory_Fence.hpp"
//----------------------------------------------------------------------------
// Provide volatile_load and safe_load