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
|
/*
* GridTools
*
* Copyright (c) 2014-2023, ETH Zurich
* All rights reserved.
*
* Please, refer to the LICENSE file in the root directory.
* SPDX-License-Identifier: BSD-3-Clause
*/
#include <gridtools/common/int_vector.hpp>
#include <type_traits>
#include <gtest/gtest.h>
#include <gridtools/common/hymap.hpp>
#include <gridtools/common/integral_constant.hpp>
#include <cuda_test_helper.hpp>
namespace gridtools {
namespace {
using namespace literals;
struct a;
struct b;
struct c;
__device__ hymap::keys<a, b, c>::values<int, long int, unsigned int> plus_device(
hymap::keys<a, b>::values<int, long int> const &m1,
hymap::keys<b, c>::values<int, unsigned int> const &m2) {
return int_vector::plus(m1, m2);
}
TEST(plus, device) {
auto m1 = hymap::keys<a, b>::make_values(1, 2l);
auto m2 = hymap::keys<b, c>::make_values(10, 20u);
auto testee = on_device::exec(GT_MAKE_INTEGRAL_CONSTANT_FROM_VALUE(&plus_device), m1, m2);
static_assert(std::is_same_v<int, std::decay_t<decltype(at_key<a>(testee))>>);
static_assert(std::is_same_v<long int, std::decay_t<decltype(at_key<b>(testee))>>);
static_assert(std::is_same_v<unsigned int, std::decay_t<decltype(at_key<c>(testee))>>);
EXPECT_EQ(1, at_key<a>(testee));
EXPECT_EQ(12, at_key<b>(testee));
EXPECT_EQ(20, at_key<c>(testee));
}
__device__ hymap::keys<a, b>::values<int, int> multiply_device(
hymap::keys<a, b>::values<integral_constant<int, 1>, int> const &vec, int scalar) {
return int_vector::multiply(vec, scalar);
}
TEST(multiply, device) {
auto vec = hymap::keys<a, b>::make_values(1_c, 2);
auto testee = on_device::exec(GT_MAKE_INTEGRAL_CONSTANT_FROM_VALUE(&multiply_device), vec, 2);
EXPECT_EQ(2, at_key<a>(testee));
EXPECT_EQ(4, at_key<b>(testee));
}
__device__ hymap::keys<a, c>::values<int, integral_constant<int, 2>> normalize_device(
hymap::keys<a, b, c>::values<int, integral_constant<int, 0>, integral_constant<int, 2>> const &vec) {
return int_vector::prune_zeros(vec);
}
TEST(prune_zeros, device) {
auto vec = hymap::keys<a, b, c>::make_values(1, 0_c, 2_c);
auto testee = on_device::exec(GT_MAKE_INTEGRAL_CONSTANT_FROM_VALUE(&normalize_device), vec);
EXPECT_EQ(1, at_key<a>(testee));
static_assert(!has_key<decltype(testee), b>());
static_assert(std::is_same_v<integral_constant<int, 2>, std::decay_t<decltype(at_key<c>(testee))>>);
}
} // namespace
} // namespace gridtools
#include "test_int_vector.cpp"
|
