Commit 63330bff authored by Paul's avatar Paul
Browse files

Rename to just readlane

parent 282fd136
......@@ -89,7 +89,7 @@ __device__ T dpp_swizzle(T& x)
}
template <unsigned int SrcLane, unsigned int Width, class T>
__device__ T dpp_readlane(T& x)
__device__ T readlane(T& x)
{
static_assert(is_power_of_2(Width), "Width must be a power of 2");
return dpp_op(x, [](auto i) { return __shfl(i, SrcLane, Width); });
......
......@@ -172,7 +172,7 @@ __device__ auto subwave_reduce(index idx, Op op, T init, Index n, F f)
idx.local_subwave_stride<SubWaveSize>(
n, [&](auto i, auto d) { x = op(x, index::invoke_loop(f, i, d)); });
dpp_reduce<SubWaveSize>(x, op);
return dpp_readlane<SubWaveSize - 1, SubWaveSize>(x);
return readlane<SubWaveSize - 1, SubWaveSize>(x);
}
template <class Op, class T, class Index, class F>
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment