-
Notifications
You must be signed in to change notification settings - Fork 8
Expand file tree
/
Copy pathmain.cpp
More file actions
52 lines (39 loc) · 1.17 KB
/
main.cpp
File metadata and controls
52 lines (39 loc) · 1.17 KB
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
#include <classes.hpp>
KOKKOS_FUNCTION
Foo::Foo() { val = 0; }
KOKKOS_FUNCTION
Foo_1::Foo_1() { val = 1; }
KOKKOS_FUNCTION
int Foo_1::value() { return val; }
KOKKOS_FUNCTION
Foo_2::Foo_2() { val = 2; }
KOKKOS_FUNCTION
int Foo_2::value() { return val; }
int main(int argc, char* argv[]) {
Kokkos::initialize(argc, argv);
{
Foo* f_1 = (Foo*)Kokkos::kokkos_malloc(sizeof(Foo_1));
Foo* f_2 = (Foo*)Kokkos::kokkos_malloc(sizeof(Foo_2));
Kokkos::parallel_for(
"CreateObjects", 1, KOKKOS_LAMBDA(const int&) {
new ((Foo_1*)f_1) Foo_1();
new ((Foo_2*)f_2) Foo_2();
});
int value_1, value_2;
Kokkos::parallel_reduce(
"CheckValues", 1,
KOKKOS_LAMBDA(const int&, int& lsum) { lsum = f_1->value(); }, value_1);
Kokkos::parallel_reduce(
"CheckValues", 1,
KOKKOS_LAMBDA(const int&, int& lsum) { lsum = f_2->value(); }, value_2);
printf("Values: %i %i\n", value_1, value_2);
Kokkos::parallel_for(
"DestroyObjects", 1, KOKKOS_LAMBDA(const int&) {
f_1->~Foo();
f_2->~Foo();
});
Kokkos::kokkos_free(f_1);
Kokkos::kokkos_free(f_2);
}
Kokkos::finalize();
}