Format

src/targets/gpu/jit/fused_reduce.cpp

@@ -126,7 +126,7 @@ struct fused_reduce_compiler : compiler<fused_reduce_compiler>
             // Vectorize if the axis is a reduction axis
             if(options.virtual_inputs.back().lens()[faxis] == 1)
                 vec = vectorize::elements(ctx, faxis, options.virtual_inputs);
-            auto relements = reduced_shape.elements() / vec.size;
+            auto relements  = reduced_shape.elements() / vec.size;
             auto block_size = compute_block_size(relements, 256);
             options.set_launch_params(
                 v, compute_global_for(ctx, nelements * block_size, 256), block_size);