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
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// SPDX-FileCopyrightText: Copyright Contributors to the Kokkos project
/// @Kokkos_Feature_Level_Required:2
// Unit test for atomic exchange, atomic add and atomic sub.
// Atomic exchange test : we interchange value1 with value2 and check for
// correctness. Atomic add test : we add value2 to value1 and check for
// correctness. Atomic sub test : we subtract value2 from value1 and check for
// correctmess.
#include <Kokkos_Macros.hpp>
#ifdef KOKKOS_ENABLE_EXPERIMENTAL_CXX20_MODULES
import kokkos.core;
#else
#include <Kokkos_Core.hpp>
#endif
#include <gtest/gtest.h>
namespace Test {
struct TestIncrAtomic {
using value_type = double;
value_type value1 = 1.5, value2 = 0.5;
void testExchange() {
value_type ret_value = Kokkos::atomic_exchange(&value1, value2);
ASSERT_EQ(value1, 0.5);
ASSERT_EQ(ret_value, 1.5);
}
void testAdd() {
Kokkos::atomic_add(&value1, value2);
ASSERT_EQ(value1, 2.0);
}
void testSub() {
Kokkos::atomic_sub(&value1, value2);
ASSERT_EQ(value1, 1.0);
}
};
TEST(TEST_CATEGORY, IncrTest_02_AtomicExchange) {
TestIncrAtomic test;
test.testExchange();
}
TEST(TEST_CATEGORY, IncrTest_02_AtomicAdd) {
TestIncrAtomic test;
test.testAdd();
}
TEST(TEST_CATEGORY, IncrTest_02_AtomicSub) {
TestIncrAtomic test;
test.testSub();
}
} // namespace Test
|
