"src/git@developer.sourcefind.cn:lacacy/qwen_lmdeploy.git" did not exist on "4903d3cc9deeece7b28024971d1279f4f085d83e"
Commit 8c18825b authored by Paul's avatar Paul
Browse files

Format

parent 2b9f612f
...@@ -125,7 +125,7 @@ struct reduce_compiler : compiler<reduce_compiler> ...@@ -125,7 +125,7 @@ struct reduce_compiler : compiler<reduce_compiler>
// Vectorize if the axis is a reduction axis // Vectorize if the axis is a reduction axis
if(options.virtual_inputs.back().lens()[faxis] == 1) if(options.virtual_inputs.back().lens()[faxis] == 1)
vec = vectorize::elements(ctx, faxis, options.virtual_inputs); vec = vectorize::elements(ctx, faxis, options.virtual_inputs);
auto relements = get_reduce_elements(options.virtual_inputs) / vec.size; auto relements = get_reduce_elements(options.virtual_inputs) / vec.size;
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, compute_global_for(ctx, nelements * block_size, 256), block_size); v, compute_global_for(ctx, nelements * block_size, 256), block_size);
......
...@@ -460,7 +460,7 @@ struct lane ...@@ -460,7 +460,7 @@ struct lane
} }
}; };
template<class Size, class F> template <class Size, class F>
constexpr inner_storage<Size, F> make_inner_storage(Size, F f) constexpr inner_storage<Size, F> make_inner_storage(Size, F f)
{ {
return {f}; return {f};
...@@ -496,9 +496,7 @@ struct lane ...@@ -496,9 +496,7 @@ struct lane
template <class R, class F, class N, class... Ts> template <class R, class F, class N, class... Ts>
__device__ auto inner_impl(F f, N n, Ts&&... xs) const __device__ auto inner_impl(F f, N n, Ts&&... xs) const
{ {
return make_inner_storage(n, [=](auto j, auto d) { return make_inner_storage(n, [=](auto j, auto d) { return f(xs(j, d)...); });
return f(xs(j, d)...);
});
} }
}; };
template <class Slicer> template <class Slicer>
......
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