File tree Expand file tree Collapse file tree 1 file changed +12
-0
lines changed
cub/cub/device/dispatch/tuning Expand file tree Collapse file tree 1 file changed +12
-0
lines changed Original file line number Diff line number Diff line change @@ -127,6 +127,18 @@ struct ReducePolicyWrapper<StaticPolicyT,
127127 CUB_DEFINE_SUB_POLICY_GETTER (SingleTile)
128128 CUB_DEFINE_SUB_POLICY_GETTER (SegmentedReduce)
129129 CUB_DEFINE_SUB_POLICY_GETTER (ReduceNondeterministic)
130+
131+ // TODO(bgruber): no longer needed by CCCL.C for reduce, but still needed for segmented_reduce
132+ #if defined(CUB_ENABLE_POLICY_PTX_JSON)
133+ _CCCL_DEVICE static constexpr auto EncodedPolicy ()
134+ {
135+ using namespace ptx_json ;
136+ return object<key<" ReducePolicy" >() = Reduce ().EncodedPolicy (),
137+ key<" SingleTilePolicy" >() = SingleTile ().EncodedPolicy (),
138+ key<" SegmentedReducePolicy" >() = SegmentedReduce ().EncodedPolicy (),
139+ key<" ReduceNondeterministicPolicy" >() = ReduceNondeterministic ().EncodedPolicy ()>();
140+ }
141+ #endif
130142};
131143
132144template <typename PolicyT>
You can’t perform that action at this time.
0 commit comments