Commit c47fa173 authored by Paul's avatar Paul
Browse files

Format

parent 50e256a1
...@@ -72,7 +72,7 @@ static shape get_reduced_shape(const shape& s, const std::vector<T>& axes) ...@@ -72,7 +72,7 @@ static shape get_reduced_shape(const shape& s, const std::vector<T>& axes)
return shape{s.type(), lens}; return shape{s.type(), lens};
} }
template<class ReduceLens> template <class ReduceLens>
static std::string get_reduce_algo(const std::vector<shape>& inputs, ReduceLens rlens) static std::string get_reduce_algo(const std::vector<shape>& inputs, ReduceLens rlens)
{ {
#if 0 #if 0
...@@ -117,7 +117,7 @@ struct fused_reduce_compiler : compiler<fused_reduce_compiler> ...@@ -117,7 +117,7 @@ struct fused_reduce_compiler : compiler<fused_reduce_compiler>
} }
auto relements = get_reduce_elements(options.virtual_inputs) / vec.size; auto relements = get_reduce_elements(options.virtual_inputs) / vec.size;
auto nelements = options.virtual_inputs.back().elements(); auto nelements = options.virtual_inputs.back().elements();
auto algo = v.get("algo", get_reduce_algo(options.virtual_inputs, reduced_shape.lens())); auto algo = v.get("algo", get_reduce_algo(options.virtual_inputs, reduced_shape.lens()));
if(algo == "block") if(algo == "block")
{ {
auto block_size = compute_block_size(relements, 256); auto block_size = compute_block_size(relements, 256);
......
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