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
75
76
77
78
79
80
81
82
83
84
85
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// SPDX-FileCopyrightText: Copyright Contributors to the Kokkos project
#include <Kokkos_Macros.hpp>
#ifdef KOKKOS_ENABLE_EXPERIMENTAL_CXX20_MODULES
import kokkos.core;
#else
#include <Kokkos_Core.hpp>
#endif
#include <gtest/gtest.h>
#include <tools/include/ToolTestingUtilities.hpp>
namespace {
template <class MemorySpace>
void test_view_bad_alloc() {
auto too_large = std::numeric_limits<size_t>::max() - 42;
std::string label = "my_label";
try {
auto should_always_fail =
Kokkos::View<double *, MemorySpace>(label, too_large);
FAIL() << "It should have thrown.";
} catch (std::runtime_error const &error) {
std::string msg = error.what();
ASSERT_PRED_FORMAT2(
::testing::IsSubstring,
std::string(MemorySpace::name()) + " memory space failed to allocate",
msg)
<< "memory space name is missing";
ASSERT_PRED_FORMAT2(::testing::IsSubstring,
std::string("(label=\"") + label + "\")", msg)
<< "label is missing";
}
}
TEST(TEST_CATEGORY, view_bad_alloc) {
using ExecutionSpace = TEST_EXECSPACE;
using MemorySpace = ExecutionSpace::memory_space;
#if defined(__has_feature)
#if __has_feature(address_sanitizer)
if (std::is_same_v<MemorySpace, Kokkos::HostSpace>) {
GTEST_SKIP() << "AddressSanitizer detects allocating too much memory "
"preventing our checks to run";
}
#endif
#endif
#if defined(KOKKOS_ENABLE_OPENACC) // FIXME_OPENACC
if (std::is_same_v<ExecutionSpace, Kokkos::Experimental::OpenACC>) {
GTEST_SKIP() << "acc_malloc() not properly returning nullptr";
}
#endif
#if defined(_WIN32) && defined(KOKKOS_ENABLE_CUDA)
if (std::is_same_v<ExecutionSpace, Kokkos::Cuda>) {
GTEST_SKIP() << "MSVC/CUDA segfaults when allocating too much memory";
}
#endif
using namespace Kokkos::Test::Tools;
listen_tool_events(Config::DisableAll(), Config::EnableAllocs());
ASSERT_TRUE(validate_absence(
[] { test_view_bad_alloc<MemorySpace>(); },
[](AllocateDataEvent) { return MatchDiagnostic{true}; }));
listen_tool_events(Config::DisableAll());
constexpr bool execution_space_is_device =
std::is_same_v<ExecutionSpace, Kokkos::DefaultExecutionSpace> &&
!std::is_same_v<Kokkos::DefaultExecutionSpace,
Kokkos::DefaultHostExecutionSpace>;
if constexpr (execution_space_is_device) {
#ifdef KOKKOS_HAS_SHARED_SPACE
test_view_bad_alloc<Kokkos::SharedSpace>();
#endif
#ifdef KOKKOS_HAS_SHARED_HOST_PINNED_SPACE
test_view_bad_alloc<Kokkos::SharedHostPinnedSpace>();
#endif
}
}
} // namespace
|
