"sgl-kernel/vscode:/vscode.git/clone" did not exist on "1a3fa75f2f4e6ea9015570e222e8150d911707b8"
Commit b3935928 authored by Paul's avatar Paul
Browse files

Formatting

parent 506a73ec
......@@ -77,11 +77,11 @@ struct hip_shape
MIGRAPHX_DEVICE_CONSTEXPR hip_index carry(hip_index result) const
{
std::ptrdiff_t rem = 0;
for(std::ptrdiff_t i = result.size()-1; i >= 0; i--)
for(std::ptrdiff_t i = result.size() - 1; i >= 0; i--)
{
auto z = result[i] + rem;
rem = z - std::ptrdiff_t(lens[i]) + 1;
if (rem > 0)
rem = z - std::ptrdiff_t(lens[i]) + 1;
if(rem > 0)
z -= rem;
else
rem = 0;
......
......@@ -70,7 +70,7 @@ void reduce_sum(hipStream_t stream, const argument& result, const argument& arg)
const std::size_t block_size = compute_block_size(relements, max_block_size);
gs_launch(stream, nelements * block_size, block_size)([=](auto i, auto idx) __device__ {
const auto out_idx = i / block_size;
auto base_idx = output.get_shape().multi(out_idx);
auto base_idx = output.get_shape().multi(out_idx);
auto r = block_reduce<max_block_size>(idx, sum{}, 0, relements, [&](auto j) __device__ {
auto reduce_idx = reduce_shape.multi(j);
return input[reduce_idx + base_idx];
......
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