229 lines
8.0 KiB
C++
229 lines
8.0 KiB
C++
/*
|
|
//@HEADER
|
|
// ************************************************************************
|
|
//
|
|
// Kokkos v. 2.0
|
|
// Copyright (2014) Sandia Corporation
|
|
//
|
|
// Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
|
|
// the U.S. Government retains certain rights in this software.
|
|
//
|
|
// Redistribution and use in source and binary forms, with or without
|
|
// modification, are permitted provided that the following conditions are
|
|
// met:
|
|
//
|
|
// 1. Redistributions of source code must retain the above copyright
|
|
// notice, this list of conditions and the following disclaimer.
|
|
//
|
|
// 2. Redistributions in binary form must reproduce the above copyright
|
|
// notice, this list of conditions and the following disclaimer in the
|
|
// documentation and/or other materials provided with the distribution.
|
|
//
|
|
// 3. Neither the name of the Corporation nor the names of the
|
|
// contributors may be used to endorse or promote products derived from
|
|
// this software without specific prior written permission.
|
|
//
|
|
// THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
|
|
// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
|
// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
|
|
// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
|
// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
|
// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
|
// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
|
// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
|
// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
|
// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
//
|
|
// Questions? Contact Christian R. Trott (crtrott@sandia.gov)
|
|
//
|
|
// ************************************************************************
|
|
//@HEADER
|
|
*/
|
|
|
|
#ifndef KOKKOS_TEST_EXPERIMENTAL_ERROR_REPORTER_HPP
|
|
#define KOKKOS_TEST_EXPERIMENTAL_ERROR_REPORTER_HPP
|
|
|
|
#include <gtest/gtest.h>
|
|
#include <iostream>
|
|
#include <Kokkos_Core.hpp>
|
|
|
|
namespace Test {
|
|
|
|
// Just save the data in the report. Informative text goies in the operator<<(..).
|
|
template <typename DataType1, typename DataType2, typename DataType3>
|
|
struct ThreeValReport
|
|
{
|
|
DataType1 m_data1;
|
|
DataType2 m_data2;
|
|
DataType3 m_data3;
|
|
|
|
};
|
|
|
|
template <typename DataType1, typename DataType2, typename DataType3>
|
|
std::ostream &operator<<(std::ostream & os, const ThreeValReport<DataType1, DataType2, DataType3> &val)
|
|
{
|
|
return os << "{" << val.m_data1 << " " << val.m_data2 << " " << val.m_data3 << "}";
|
|
}
|
|
|
|
template<typename ReportType>
|
|
void checkReportersAndReportsAgree(const std::vector<int> &reporters,
|
|
const std::vector<ReportType> &reports)
|
|
{
|
|
for (size_t i = 0; i < reports.size(); ++i) {
|
|
EXPECT_EQ(1, reporters[i] % 2);
|
|
EXPECT_EQ(reporters[i], reports[i].m_data1);
|
|
}
|
|
}
|
|
|
|
|
|
template <typename DeviceType>
|
|
struct ErrorReporterDriverBase {
|
|
|
|
typedef ThreeValReport<int, int, double> report_type;
|
|
typedef Kokkos::Experimental::ErrorReporter<report_type, DeviceType> error_reporter_type;
|
|
error_reporter_type m_errorReporter;
|
|
|
|
ErrorReporterDriverBase(int reporter_capacity, int test_size)
|
|
: m_errorReporter(reporter_capacity) { }
|
|
|
|
KOKKOS_INLINE_FUNCTION bool error_condition(const int work_idx) const { return (work_idx % 2 != 0); }
|
|
|
|
void check_expectations(int reporter_capacity, int test_size)
|
|
{
|
|
int num_reported = m_errorReporter.getNumReports();
|
|
int num_attempts = m_errorReporter.getNumReportAttempts();
|
|
|
|
int expected_num_reports = std::min(reporter_capacity, test_size / 2);
|
|
EXPECT_EQ(expected_num_reports, num_reported);
|
|
EXPECT_EQ(test_size / 2, num_attempts);
|
|
|
|
bool expect_full = (reporter_capacity <= (test_size / 2));
|
|
bool reported_full = m_errorReporter.full();
|
|
EXPECT_EQ(expect_full, reported_full);
|
|
}
|
|
};
|
|
|
|
template <typename ErrorReporterDriverType>
|
|
void TestErrorReporter()
|
|
{
|
|
typedef ErrorReporterDriverType tester_type;
|
|
std::vector<int> reporters;
|
|
std::vector<typename tester_type::report_type> reports;
|
|
|
|
tester_type test1(100, 10);
|
|
test1.m_errorReporter.getReports(reporters, reports);
|
|
checkReportersAndReportsAgree(reporters, reports);
|
|
|
|
tester_type test2(10, 100);
|
|
test2.m_errorReporter.getReports(reporters, reports);
|
|
checkReportersAndReportsAgree(reporters, reports);
|
|
|
|
typename Kokkos::View<int*, typename ErrorReporterDriverType::execution_space >::HostMirror view_reporters;
|
|
typename Kokkos::View<typename tester_type::report_type*, typename ErrorReporterDriverType::execution_space >::HostMirror
|
|
view_reports;
|
|
test2.m_errorReporter.getReports(view_reporters, view_reports);
|
|
|
|
int num_reports = view_reporters.extent(0);
|
|
reporters.clear();
|
|
reports.clear();
|
|
reporters.reserve(num_reports);
|
|
reports.reserve(num_reports);
|
|
|
|
for (int i = 0; i < num_reports; ++i) {
|
|
reporters.push_back(view_reporters(i));
|
|
reports.push_back(view_reports(i));
|
|
}
|
|
checkReportersAndReportsAgree(reporters, reports);
|
|
|
|
}
|
|
|
|
|
|
template <typename DeviceType>
|
|
struct ErrorReporterDriver : public ErrorReporterDriverBase<DeviceType>
|
|
{
|
|
typedef ErrorReporterDriverBase<DeviceType> driver_base;
|
|
typedef typename driver_base::error_reporter_type::execution_space execution_space;
|
|
|
|
ErrorReporterDriver(int reporter_capacity, int test_size)
|
|
: driver_base(reporter_capacity, test_size)
|
|
{
|
|
execute(reporter_capacity, test_size);
|
|
|
|
// Test that clear() and resize() work across memory spaces.
|
|
if (reporter_capacity < test_size) {
|
|
driver_base::m_errorReporter.clear();
|
|
driver_base::m_errorReporter.resize(test_size);
|
|
execute(test_size, test_size);
|
|
}
|
|
}
|
|
|
|
void execute(int reporter_capacity, int test_size)
|
|
{
|
|
Kokkos::parallel_for(Kokkos::RangePolicy<execution_space>(0,test_size), *this);
|
|
driver_base::check_expectations(reporter_capacity, test_size);
|
|
}
|
|
|
|
KOKKOS_INLINE_FUNCTION
|
|
void operator()(const int work_idx) const
|
|
{
|
|
if (driver_base::error_condition(work_idx)) {
|
|
double val = M_PI * static_cast<double>(work_idx);
|
|
typename driver_base::report_type report = {work_idx, -2*work_idx, val};
|
|
driver_base::m_errorReporter.add_report(work_idx, report);
|
|
}
|
|
}
|
|
};
|
|
|
|
#if defined(KOKKOS_CLASS_LAMBDA)
|
|
template <typename DeviceType>
|
|
struct ErrorReporterDriverUseLambda : public ErrorReporterDriverBase<DeviceType>
|
|
{
|
|
|
|
typedef ErrorReporterDriverBase<DeviceType> driver_base;
|
|
typedef typename driver_base::error_reporter_type::execution_space execution_space;
|
|
|
|
ErrorReporterDriverUseLambda(int reporter_capacity, int test_size)
|
|
: driver_base(reporter_capacity, test_size)
|
|
{
|
|
Kokkos::parallel_for(Kokkos::RangePolicy<execution_space>(0,test_size), KOKKOS_CLASS_LAMBDA (const int work_idx) {
|
|
if (driver_base::error_condition(work_idx)) {
|
|
double val = M_PI * static_cast<double>(work_idx);
|
|
typename driver_base::report_type report = {work_idx, -2*work_idx, val};
|
|
driver_base::m_errorReporter.add_report(work_idx, report);
|
|
}
|
|
});
|
|
driver_base::check_expectations(reporter_capacity, test_size);
|
|
}
|
|
|
|
};
|
|
#endif
|
|
|
|
|
|
#ifdef KOKKOS_ENABLE_OPENMP
|
|
struct ErrorReporterDriverNativeOpenMP : public ErrorReporterDriverBase<Kokkos::OpenMP>
|
|
{
|
|
typedef ErrorReporterDriverBase<Kokkos::OpenMP> driver_base;
|
|
typedef typename driver_base::error_reporter_type::execution_space execution_space;
|
|
|
|
ErrorReporterDriverNativeOpenMP(int reporter_capacity, int test_size)
|
|
: driver_base(reporter_capacity, test_size)
|
|
{
|
|
#pragma omp parallel for
|
|
for(int work_idx = 0; work_idx < test_size; ++work_idx)
|
|
{
|
|
if (driver_base::error_condition(work_idx)) {
|
|
double val = M_PI * static_cast<double>(work_idx);
|
|
typename driver_base::report_type report = {work_idx, -2*work_idx, val};
|
|
driver_base::m_errorReporter.add_report(work_idx, report);
|
|
}
|
|
};
|
|
driver_base::check_expectations(reporter_capacity, test_size);
|
|
}
|
|
};
|
|
#endif
|
|
|
|
} // namespace Test
|
|
#endif // #ifndef KOKKOS_TEST_ERROR_REPORTING_HPP
|
|
|