Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F121847531
TestErrorReporter.hpp
No One
Temporary
Actions
Download File
Edit File
Delete File
View Transforms
Subscribe
Mute Notifications
Award Token
Subscribers
None
File Metadata
Details
File Info
Storage
Attached
Created
Mon, Jul 14, 09:26
Size
8 KB
Mime Type
text/x-c++
Expires
Wed, Jul 16, 09:26 (1 d, 23 h)
Engine
blob
Format
Raw Data
Handle
27400838
Attached To
rLAMMPS lammps
TestErrorReporter.hpp
View Options
/*
//@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 H. Carter Edwards (hcedwar@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_HAVE_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
Event Timeline
Log In to Comment