1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
//@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_SCRATCH_HPP
#define KOKKOS_TEST_TEAM_SCRATCH_HPP
#include <TestTeam.hpp>
namespace Test {
TEST(TEST_CATEGORY, team_shared_request) {
TestSharedTeam<TEST_EXECSPACE, Kokkos::Schedule<Kokkos::Static> >();
TestSharedTeam<TEST_EXECSPACE, Kokkos::Schedule<Kokkos::Dynamic> >();
}
TEST(TEST_CATEGORY, team_scratch_request) {
TestScratchTeam<TEST_EXECSPACE, Kokkos::Schedule<Kokkos::Static> >();
TestScratchTeam<TEST_EXECSPACE, Kokkos::Schedule<Kokkos::Dynamic> >();
}
TEST(TEST_CATEGORY, team_lambda_shared_request) {
TestLambdaSharedTeam<Kokkos::HostSpace, TEST_EXECSPACE,
Kokkos::Schedule<Kokkos::Static> >();
TestLambdaSharedTeam<Kokkos::HostSpace, TEST_EXECSPACE,
Kokkos::Schedule<Kokkos::Dynamic> >();
}
TEST(TEST_CATEGORY, scratch_align) { TestScratchAlignment<TEST_EXECSPACE>(); }
TEST(TEST_CATEGORY, shmem_size) { TestShmemSize<TEST_EXECSPACE>(); }
TEST(TEST_CATEGORY, multi_level_scratch) {
// FIXME_OPENMPTARGET This unit test needs ~350KB of scratch memory for L0 and
// L1 combined per team. Currently OpenMPTarget cannot allocate this high
// amount of scratch memory.
#if !defined(KOKKOS_ENABLE_OPENMPTARGET)
TestMultiLevelScratchTeam<TEST_EXECSPACE,
Kokkos::Schedule<Kokkos::Static> >();
TestMultiLevelScratchTeam<TEST_EXECSPACE,
Kokkos::Schedule<Kokkos::Dynamic> >();
#endif
}
struct DummyTeamParallelForFunctor {
KOKKOS_FUNCTION void operator()(
Kokkos::TeamPolicy<TEST_EXECSPACE>::member_type) const {}
};
TEST(TEST_CATEGORY, team_scratch_memory_index_parallel_for) {
// Requesting per team scratch memory for a largish number of teams, resulted
// in problems computing the correct scratch pointer due to missed
// initialization of the maximum number of scratch pad indices in the Cuda
// baackend.
const int scratch_size = 4896;
const int league_size = 7535;
Kokkos::TeamPolicy<TEST_EXECSPACE> policy(league_size, Kokkos::AUTO);
policy.set_scratch_size(1, Kokkos::PerTeam(scratch_size));
Kokkos::parallel_for("kernel", policy, DummyTeamParallelForFunctor());
}
} // namespace Test
#endif
|
