Commit db2def39 authored by Paul's avatar Paul
Browse files

Format

parent f1f60be1
......@@ -30,7 +30,7 @@ vectorize vectorize::elements(std::size_t axis, const std::vector<shape>& inputs
auto len = input.lens()[axis];
if(stride != 0 and stride != 1)
return 1;
if (len == 1)
if(len == 1)
return sizes.front();
auto it = std::find_if(
sizes.begin(), sizes.end(), [&](auto i) { return (len % i) == 0; });
......
......@@ -115,14 +115,11 @@ struct reduce_compiler : compiler<reduce_compiler>
{
auto block_size = compute_block_size(relements, 256);
options.set_launch_params(
v,
compute_global_for(ctx, nelements * block_size, 256),
block_size);
v, compute_global_for(ctx, nelements * block_size, 256), block_size);
}
else if(algo == "lane")
{
options.set_launch_params(
v, compute_global_for(ctx, nelements, 256));
options.set_launch_params(v, compute_global_for(ctx, nelements, 256));
}
else
{
......
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