Open shwina opened 1 day ago
I was under the impression that d_in
/d_out
could be iterators too, in which case reduce_info
would need to know them (as part of the problem definition) for later codegen?
I was under the impression that
d_in
/d_out
could be iterators too,
Yes. The API I have right now (in #2788) is hack-ish, just enough for full testing. I want to discuss with @shwina (today) what the iterator API should look like.
Currently,
reduce_into
usage looks like:Note that the initial construction of
reducer
shouldn't strictly need the argumentsd_in
,d_out
,h_init
. In fact, passing any placeholder arrays of the same data type will serve the same purpose.We should refactor such that only the required information is passed into the constructor.
Additional Context
https://github.com/NVIDIA/cccl/pull/3001/#discussion_r1866733390