Updating Kokkos lib
This commit is contained in:
@ -56,7 +56,7 @@
|
||||
#include <Kokkos_CudaSpace.hpp>
|
||||
|
||||
#include <Kokkos_Parallel.hpp>
|
||||
#include <Kokkos_TaskPolicy.hpp>
|
||||
#include <Kokkos_TaskScheduler.hpp>
|
||||
#include <Kokkos_Layout.hpp>
|
||||
#include <Kokkos_ScratchSpace.hpp>
|
||||
#include <Kokkos_MemoryTraits.hpp>
|
||||
@ -229,6 +229,39 @@ private:
|
||||
namespace Kokkos {
|
||||
namespace Impl {
|
||||
|
||||
template<>
|
||||
struct MemorySpaceAccess
|
||||
< Kokkos::CudaSpace
|
||||
, Kokkos::Cuda::scratch_memory_space
|
||||
>
|
||||
{
|
||||
enum { assignable = false };
|
||||
enum { accessible = true };
|
||||
enum { deepcopy = false };
|
||||
};
|
||||
|
||||
#if defined( KOKKOS_USE_CUDA_UVM )
|
||||
|
||||
// If forcing use of UVM everywhere
|
||||
// then must assume that CudaUVMSpace
|
||||
// can be a stand-in for CudaSpace.
|
||||
// This will fail when a strange host-side execution space
|
||||
// that defines CudaUVMSpace as its preferredmemory space.
|
||||
|
||||
template<>
|
||||
struct MemorySpaceAccess
|
||||
< Kokkos::CudaUVMSpace
|
||||
, Kokkos::Cuda::scratch_memory_space
|
||||
>
|
||||
{
|
||||
enum { assignable = false };
|
||||
enum { accessible = true };
|
||||
enum { deepcopy = false };
|
||||
};
|
||||
|
||||
#endif
|
||||
|
||||
|
||||
template<>
|
||||
struct VerifyExecutionCanAccessMemorySpace
|
||||
< Kokkos::CudaSpace
|
||||
@ -259,9 +292,6 @@ struct VerifyExecutionCanAccessMemorySpace
|
||||
|
||||
#include <Cuda/Kokkos_CudaExec.hpp>
|
||||
#include <Cuda/Kokkos_Cuda_View.hpp>
|
||||
|
||||
#include <Cuda/KokkosExp_Cuda_View.hpp>
|
||||
|
||||
#include <Cuda/Kokkos_Cuda_Parallel.hpp>
|
||||
#include <Cuda/Kokkos_Cuda_Task.hpp>
|
||||
|
||||
|
||||
Reference in New Issue
Block a user