Shuenhoy/vtk-m
1
0
Fork 0
mirror of https://gitlab.kitware.com/vtk/vtk-m synced 2024-09-19 18:45:43 +00:00
Code Issues 2 Actions Packages Projects Releases Wiki Activity

DispatcherReduceByKey uses the scheduling_range like all other dispatchers

Browse Source
This commit is contained in:
Robert Maynard 2019-05-15 11:42:02 -04:00
parent bd626dac79
commit 63fe0f0965
2 changed files with 18 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
vtkm/worklet/DispatcherReduceByKey.h
View File

@ -59,8 +59,8 @@ public:
const InputDomainType& inputDomain = invocation.GetInputDomain();
// Now that we have the input domain, we can extract the range of the
// scheduling and call BadicInvoke.
this->BasicInvoke(invocation, inputDomain.GetInputRange());
// scheduling and call BasicInvoke.
this->BasicInvoke(invocation, internal::scheduling_range(inputDomain));
}
};
}

16
vtkm/worklet/internal/DispatcherBase.h
View File

@ -48,6 +48,8 @@ namespace vtkm
{
namespace worklet
{
template <typename T>
class VTKM_ALWAYS_EXPORT Keys;
namespace internal
{
@ -57,6 +59,13 @@ inline auto scheduling_range(const Domain& inputDomain) -> decltype(inputDomain.
return inputDomain.GetNumberOfValues();
}
template <typename KeyType>
inline auto scheduling_range(const vtkm::worklet::Keys<KeyType>& inputDomain)
-> decltype(inputDomain.GetInputRange())
{
return inputDomain.GetInputRange();
}
template <typename Domain>
inline auto scheduling_range(const Domain* const inputDomain)
-> decltype(inputDomain->GetNumberOfValues())
@ -64,6 +73,13 @@ inline auto scheduling_range(const Domain* const inputDomain)
return inputDomain->GetNumberOfValues();
}
template <typename KeyType>
inline auto scheduling_range(const vtkm::worklet::Keys<KeyType>* const inputDomain)
-> decltype(inputDomain->GetInputRange())
{
return inputDomain->GetInputRange();
}
template <typename Domain, typename SchedulingRangeType>
inline auto scheduling_range(const Domain& inputDomain, SchedulingRangeType type)
-> decltype(inputDomain.GetSchedulingRange(type))