125 lines
4.4 KiB
C++
125 lines
4.4 KiB
C++
//@HEADER
|
|
// ************************************************************************
|
|
//
|
|
// Kokkos v. 4.0
|
|
// Copyright (2022) National Technology & Engineering
|
|
// Solutions of Sandia, LLC (NTESS).
|
|
//
|
|
// Under the terms of Contract DE-NA0003525 with NTESS,
|
|
// the U.S. Government retains certain rights in this software.
|
|
//
|
|
// Part of Kokkos, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://kokkos.org/LICENSE for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//@HEADER
|
|
|
|
#ifndef KOKKOS_TEST_TEAM_REDUCTION_SCAN_HPP
|
|
#define KOKKOS_TEST_TEAM_REDUCTION_SCAN_HPP
|
|
#include <TestTeam.hpp>
|
|
|
|
namespace Test {
|
|
|
|
TEST(TEST_CATEGORY, team_reduction_scan) {
|
|
TestScanTeam<TEST_EXECSPACE, Kokkos::Schedule<Kokkos::Static> >(0);
|
|
TestScanTeam<TEST_EXECSPACE, Kokkos::Schedule<Kokkos::Dynamic> >(0);
|
|
TestScanTeam<TEST_EXECSPACE, Kokkos::Schedule<Kokkos::Static> >(10);
|
|
TestScanTeam<TEST_EXECSPACE, Kokkos::Schedule<Kokkos::Dynamic> >(10);
|
|
TestScanTeam<TEST_EXECSPACE, Kokkos::Schedule<Kokkos::Static> >(10000);
|
|
TestScanTeam<TEST_EXECSPACE, Kokkos::Schedule<Kokkos::Dynamic> >(10000);
|
|
}
|
|
|
|
TEST(TEST_CATEGORY, team_long_reduce) {
|
|
#ifdef KOKKOS_ENABLE_OPENMPTARGET // FIXME_OPENMPTARGET: Not implemented
|
|
if constexpr (!std::is_same<TEST_EXECSPACE,
|
|
Kokkos::Experimental::OpenMPTarget>::value)
|
|
#endif
|
|
{
|
|
TestReduceTeam<long, TEST_EXECSPACE, Kokkos::Schedule<Kokkos::Static> >(0);
|
|
TestReduceTeam<long, TEST_EXECSPACE, Kokkos::Schedule<Kokkos::Dynamic> >(0);
|
|
TestReduceTeam<long, TEST_EXECSPACE, Kokkos::Schedule<Kokkos::Static> >(3);
|
|
TestReduceTeam<long, TEST_EXECSPACE, Kokkos::Schedule<Kokkos::Dynamic> >(3);
|
|
TestReduceTeam<long, TEST_EXECSPACE, Kokkos::Schedule<Kokkos::Static> >(
|
|
100000);
|
|
TestReduceTeam<long, TEST_EXECSPACE, Kokkos::Schedule<Kokkos::Dynamic> >(
|
|
100000);
|
|
}
|
|
}
|
|
|
|
TEST(TEST_CATEGORY, team_double_reduce) {
|
|
#ifdef KOKKOS_ENABLE_OPENMPTARGET // FIXME_OPENMPTARGET: Not implemented
|
|
if constexpr (!std::is_same<TEST_EXECSPACE,
|
|
Kokkos::Experimental::OpenMPTarget>::value)
|
|
#endif
|
|
{
|
|
TestReduceTeam<double, TEST_EXECSPACE, Kokkos::Schedule<Kokkos::Static> >(
|
|
0);
|
|
TestReduceTeam<double, TEST_EXECSPACE, Kokkos::Schedule<Kokkos::Dynamic> >(
|
|
0);
|
|
TestReduceTeam<double, TEST_EXECSPACE, Kokkos::Schedule<Kokkos::Static> >(
|
|
3);
|
|
TestReduceTeam<double, TEST_EXECSPACE, Kokkos::Schedule<Kokkos::Dynamic> >(
|
|
3);
|
|
TestReduceTeam<double, TEST_EXECSPACE, Kokkos::Schedule<Kokkos::Static> >(
|
|
100000);
|
|
TestReduceTeam<double, TEST_EXECSPACE, Kokkos::Schedule<Kokkos::Dynamic> >(
|
|
100000);
|
|
}
|
|
}
|
|
|
|
template <typename ExecutionSpace>
|
|
struct DummyTeamReductionFunctor {
|
|
using TeamPolicy = Kokkos::TeamPolicy<ExecutionSpace>;
|
|
using TeamHandleType = typename TeamPolicy::member_type;
|
|
|
|
KOKKOS_FUNCTION void operator()(const TeamHandleType&, double&) const {}
|
|
};
|
|
|
|
template <typename ExecutionSpace>
|
|
void test_team_parallel_reduce(const int num_loop_size) {
|
|
using TeamPolicy = Kokkos::TeamPolicy<ExecutionSpace>;
|
|
|
|
using ReducerType = Kokkos::Sum<double>;
|
|
double result = 10.;
|
|
ReducerType reducer(result);
|
|
|
|
const int bytes_per_team = 0;
|
|
const int bytes_per_thread = 117;
|
|
|
|
TeamPolicy team_exec(num_loop_size, Kokkos::AUTO);
|
|
team_exec.set_scratch_size(1, Kokkos::PerTeam(bytes_per_team),
|
|
Kokkos::PerThread(bytes_per_thread));
|
|
|
|
Kokkos::parallel_reduce(team_exec,
|
|
DummyTeamReductionFunctor<ExecutionSpace>{}, reducer);
|
|
ASSERT_EQ(result, 0.);
|
|
}
|
|
|
|
TEST(TEST_CATEGORY, team_parallel_dummy_with_reducer_and_scratch_space) {
|
|
#ifdef KOKKOS_ENABLE_OPENMPTARGET // FIXME_OPENMPTARGET: Not implemented
|
|
if constexpr (!std::is_same<TEST_EXECSPACE,
|
|
Kokkos::Experimental::OpenMPTarget>::value)
|
|
#endif
|
|
{
|
|
test_team_parallel_reduce<TEST_EXECSPACE>(0);
|
|
test_team_parallel_reduce<TEST_EXECSPACE>(1);
|
|
}
|
|
}
|
|
|
|
TEST(TEST_CATEGORY, repeated_team_reduce) {
|
|
#ifdef KOKKOS_ENABLE_OPENMPTARGET
|
|
if (std::is_same<TEST_EXECSPACE, Kokkos::Experimental::OpenMPTarget>::value)
|
|
GTEST_SKIP() << "skipping since team_reduce for OpenMPTarget is not "
|
|
"properly implemented";
|
|
#endif
|
|
|
|
#ifdef KOKKOS_IMPL_32BIT
|
|
GTEST_SKIP() << "Failing KOKKOS_IMPL_32BIT"; // FIXME_32BIT
|
|
#endif
|
|
|
|
TestRepeatedTeamReduce<TEST_EXECSPACE>();
|
|
}
|
|
|
|
} // namespace Test
|
|
#endif
|