9#include <Compadre_Config.h>
16#ifdef COMPADRE_USE_MPI
20#include <Kokkos_Timer.hpp>
21#include <Kokkos_Core.hpp>
28int main (
int argc,
char* args[]) {
30#ifdef COMPADRE_USE_MPI
32MPI_Init(&argc, &args);
36Kokkos::initialize(argc, args);
39bool all_passed =
true;
47 if (args[2] != NULL) {
48 auto arg3toi = atoi(args[2]);
49 if (isdigit(arg3toi)) {
60 if (args[1] != NULL) {
61 auto arg2toi = atoi(args[1]);
62 if (isdigit(arg2toi)) {
73 Kokkos::Profiling::pushRegion(
"Setup Data");
76 Kokkos::View<int**, Kokkos::DefaultExecutionSpace> data_device(
"data", num_flags, num_cols);
77 Kokkos::View<int**>::HostMirror data = Kokkos::create_mirror_view(data_device);
80 Kokkos::View<int*, Kokkos::DefaultExecutionSpace> flags_device(
"flags", num_flags);
81 Kokkos::View<int*>::HostMirror flags = Kokkos::create_mirror_view(flags_device);
85 Kokkos::Profiling::popRegion();
86 Kokkos::Profiling::pushRegion(
"Filter And Extract Data");
90 for (
int i=0; i<num_flags; i++) {
91 for (
int j=0; j<num_cols; j++) {
100 Kokkos::deep_copy(data_device, data);
103 int num_filtered_flags = 0;
104 for (
int i=0; i<num_flags; i++) {
107 num_filtered_flags++;
113 Kokkos::deep_copy(flags_device, flags);
116 auto filtered_flags = filterViewByID<Kokkos::HostSpace>(flags_device, 1);
117 auto extracted_data = Extract::extractViewByIndex<Kokkos::HostSpace>(data_device, filtered_flags);
121 Kokkos::Profiling::popRegion();
122 Kokkos::Profiling::pushRegion(
"Check Filtered And Extracted Data");
126 if (filtered_flags.extent(0) != (
size_t)num_filtered_flags) {
128 std::cout <<
"Failed - number of filtered flags not matched!" << filtered_flags.extent(0) <<
" " << num_filtered_flags << std::endl;
130 for (
size_t i=0; i<filtered_flags.extent(0); i++) {
131 if (filtered_flags(i) % 2 != 0) {
133 std::cout <<
"Failed - incorrect filtered flags " << filtered_flags(i) << std::endl;
137 for (
size_t i=0; i<extracted_data.extent(0); i++) {
138 for (
size_t j=0; j<extracted_data.extent(1); j++) {
139 if (extracted_data(i, j) != 1) {
141 std::cout <<
"Failed - incorrect values in extracted view at index " << i <<
" " << j <<
" " << extracted_data(i, j) << std::endl;
148 Kokkos::Profiling::popRegion();
156#ifdef COMPADRE_USE_MPI
162 fprintf(stdout,
"Passed test \n");
165 fprintf(stdout,
"Failed test \n");
int main(int argc, char *args[])
[Parse Command Line Arguments]