Files
lammps/lib/kokkos/example/virtual_functions/main.cpp
2023-03-03 09:22:33 -07:00

55 lines
1.5 KiB
C++

//@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
#include <classes.hpp>
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();
}