Updating Kokkos lib
This commit is contained in:
@ -499,7 +499,7 @@ struct TestSharedTeam {
|
||||
|
||||
namespace Test {
|
||||
|
||||
#if defined (KOKKOS_HAVE_CXX11_DISPATCH_LAMBDA)
|
||||
#if defined (KOKKOS_ENABLE_CXX11_DISPATCH_LAMBDA)
|
||||
template< class MemorySpace, class ExecSpace, class ScheduleType >
|
||||
struct TestLambdaSharedTeam {
|
||||
|
||||
@ -519,7 +519,7 @@ struct TestLambdaSharedTeam {
|
||||
|
||||
const int SHARED_COUNT = 1000;
|
||||
int team_size = 1;
|
||||
#ifdef KOKKOS_HAVE_CUDA
|
||||
#ifdef KOKKOS_ENABLE_CUDA
|
||||
if(std::is_same<ExecSpace,Kokkos::Cuda>::value)
|
||||
team_size = 128;
|
||||
#endif
|
||||
@ -841,7 +841,7 @@ struct ClassWithShmemSizeFunction {
|
||||
|
||||
template< class ExecSpace, class ScheduleType >
|
||||
void test_team_mulit_level_scratch_test_lambda() {
|
||||
#ifdef KOKKOS_HAVE_CXX11_DISPATCH_LAMBDA
|
||||
#ifdef KOKKOS_ENABLE_CXX11_DISPATCH_LAMBDA
|
||||
Kokkos::View<int,ExecSpace,Kokkos::MemoryTraits<Kokkos::Atomic> > errors;
|
||||
Kokkos::View<int,ExecSpace> d_errors("Errors");
|
||||
errors = d_errors;
|
||||
@ -885,7 +885,7 @@ struct TestMultiLevelScratchTeam {
|
||||
|
||||
void run()
|
||||
{
|
||||
#ifdef KOKKOS_HAVE_CXX11_DISPATCH_LAMBDA
|
||||
#ifdef KOKKOS_ENABLE_CXX11_DISPATCH_LAMBDA
|
||||
Test::test_team_mulit_level_scratch_test_lambda<ExecSpace, ScheduleType>();
|
||||
#endif
|
||||
Test::ClassNoShmemSizeFunction<ExecSpace, ScheduleType> c1;
|
||||
|
||||
Reference in New Issue
Block a user