Shuenhoy/vtk-m2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add a test to use Reduce and ArrayHandleZip together.

Browse Source
This commit is contained in:
Robert Maynard 2015-06-11 15:23:29 -04:00
parent 2a2159b1e1
commit 791e0c477c
1 changed files with 36 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
vtkm/cont/testing/TestingDeviceAdapter.h
View File

@ -54,8 +54,8 @@ namespace testing {
namespace comparison {
struct SortLess
{
template<typename T, typename U>
VTKM_EXEC_CONT_EXPORT bool operator()(const T& a, const U& b) const
template<typename T>
VTKM_EXEC_CONT_EXPORT bool operator()(const T& a, const T& b) const
{
return a < b;
}
@ -76,8 +76,8 @@ struct SortLess
struct SortGreater
{
template<typename T, typename U>
VTKM_EXEC_CONT_EXPORT bool operator()(const T& a, const U& b) const
template<typename T>
VTKM_EXEC_CONT_EXPORT bool operator()(const T& a, const T& b) const
{
return a > b;
}
@ -946,6 +946,7 @@ private:
std::cout << "-------------------------------------------" << std::endl;
std::cout << "Testing Reduce" << std::endl;
{
//construct the index array
IdArrayHandle array;
Algorithm::Schedule(
@ -966,6 +967,37 @@ private:
VTKM_TEST_ASSERT(reduce_sum == inclusive_sum,
"Got different sums from Reduce and ScanInclusive");
}
std::cout << "-------------------------------------------" << std::endl;
std::cout << "Testing Reduce with ArrayHandleZip" << std::endl;
{
IdArrayHandle keys, values;
Algorithm::Schedule(
ClearArrayKernel(keys.PrepareForOutput(ARRAY_SIZE,
DeviceAdapterTag())),
ARRAY_SIZE);
Algorithm::Schedule(
ClearArrayKernel(values.PrepareForOutput(ARRAY_SIZE,
DeviceAdapterTag())),
ARRAY_SIZE);
vtkm::cont::ArrayHandleZip< IdArrayHandle,
IdArrayHandle > zipped(keys,values);
//the output of reduce and scan inclusive should be the same
typedef vtkm::Pair<vtkm::Id,vtkm::Id> ResultType;
ResultType reduce_sum_with_intial_value =
Algorithm::Reduce(zipped, ResultType(ARRAY_SIZE,ARRAY_SIZE));
ResultType expectedResult(OFFSET * ARRAY_SIZE + ARRAY_SIZE,
OFFSET * ARRAY_SIZE + ARRAY_SIZE);
VTKM_TEST_ASSERT( ( reduce_sum_with_intial_value == expectedResult),
"Got bad sum from Reduce with initial value");
}
}
static VTKM_CONT_EXPORT void TestReduceWithComparisonObject()