Skip to content

Commit 79c25a9

Browse files
Fix segmented_reduce
1 parent 4ab144a commit 79c25a9

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

cub/cub/device/dispatch/tuning/tuning_reduce.cuh

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff 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

132144
template <typename PolicyT>

0 commit comments

Comments
 (0)