Skip to content

GitLab

Commit 6570087f authored by Paul's avatar Paul
Browse files

Format

parent c2263671
Show whitespace changes
Inline Side-by-side
Showing with 15 additions and 14 deletions
src/targets/gpu/jit/concat.cpp
View file @ 6570087f
...@@ -83,7 +83,8 @@ struct concat_compiler : compiler<concat_compiler> ...@@ -83,7 +83,8 @@ struct concat_compiler : compiler<concat_compiler>
auto vec = vectorize::elements(axis, options.inputs); auto vec = vectorize::elements(axis, options.inputs);
options.set_launch_params( options.set_launch_params(
v, compute_global_for(ctx, get_concat_elements(options.inputs) / vec.size, 256)); v, compute_global_for(ctx, get_concat_elements(options.inputs) / vec.size, 256));
auto src = interpolate_string(concat_kernel, auto src = interpolate_string(
concat_kernel,
{{"kernel", options.kernel_name}, {{"kernel", options.kernel_name},
{"params", enum_params(inputs.size(), "void * private_p")}, {"params", enum_params(inputs.size(), "void * private_p")},
{"args", enum_params(inputs.size(), "private_p")}, {"args", enum_params(inputs.size(), "private_p")},
......
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