1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#pragma once
#include <vespa/eval/eval/nested_loop.h>
#include <vespa/eval/eval/value_type.h>
#include <vespa/eval/eval/interpreted_function.h>
#include <vespa/vespalib/stllike/string.h>
#include <vespa/vespalib/util/small_vector.h>
namespace vespalib::eval { struct ValueBuilderFactory; }
namespace vespalib::eval::instruction {
struct GenericConcat {
static InterpretedFunction::Instruction
make_instruction(const ValueType &result_type,
const ValueType &lhs_type, const ValueType &rhs_type,
const vespalib::string &dimension,
const ValueBuilderFactory &factory, Stash &stash);
};
struct DenseConcatPlan {
size_t right_offset;
size_t output_size;
struct InOutLoop {
size_t input_size;
SmallVector<size_t> in_loop_cnt;
SmallVector<size_t> in_stride;
SmallVector<size_t> out_stride;
// returns computed concat offset and output size
std::pair<size_t, size_t> fill_from(const ValueType &in_type,
std::string concat_dimension,
const ValueType &out_type);
template <typename F> void execute(size_t in_off, size_t out_off, const F &f) const {
run_nested_loop(in_off, out_off, in_loop_cnt, in_stride, out_stride, f);
}
~InOutLoop();
};
InOutLoop left;
InOutLoop right;
DenseConcatPlan(const ValueType &lhs_type, const ValueType &rhs_type, std::string concat_dimension, const ValueType &res_type);
~DenseConcatPlan();
};
} // namespace
|