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