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) ...@@ -89,7 +89,7 @@ __device__ T dpp_swizzle(T& x)
} }
template <unsigned int SrcLane, unsigned int Width, class T> 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"); 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); }); 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) ...@@ -172,7 +172,7 @@ __device__ auto subwave_reduce(index idx, Op op, T init, Index n, F f)
idx.local_subwave_stride<SubWaveSize>( idx.local_subwave_stride<SubWaveSize>(
n, [&](auto i, auto d) { x = op(x, index::invoke_loop(f, i, d)); }); n, [&](auto i, auto d) { x = op(x, index::invoke_loop(f, i, d)); });
dpp_reduce<SubWaveSize>(x, op); 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> 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