diff options
author | Arne Juul <arnej@verizonmedia.com> | 2021-01-20 11:49:26 +0000 |
---|---|---|
committer | Arne Juul <arnej@verizonmedia.com> | 2021-01-20 11:49:26 +0000 |
commit | 54d51a45c008773c3b65da62969b4c40bce3f52d (patch) | |
tree | 331ab0c3351b3ba0e5d6abce800abd1377907e73 /eval | |
parent | 4b85b4e3922407bbe370e04deacdbfa9aacaea80 (diff) |
minor cleanup
Diffstat (limited to 'eval')
-rw-r--r-- | eval/src/vespa/eval/instruction/mixed_inner_product_function.cpp | 16 | ||||
-rw-r--r-- | eval/src/vespa/eval/instruction/mixed_inner_product_function.h | 4 |
2 files changed, 9 insertions, 11 deletions
diff --git a/eval/src/vespa/eval/instruction/mixed_inner_product_function.cpp b/eval/src/vespa/eval/instruction/mixed_inner_product_function.cpp index 1d5d446ec59..7a44a8902c9 100644 --- a/eval/src/vespa/eval/instruction/mixed_inner_product_function.cpp +++ b/eval/src/vespa/eval/instruction/mixed_inner_product_function.cpp @@ -51,21 +51,19 @@ template <typename MCT, typename VCT, typename OCT> void my_mixed_inner_product_op(InterpretedFunction::State &state, uint64_t param_in) { const auto ¶m = unwrap_param<MixedInnerProductParam>(param_in); const auto &mixed = state.peek(1); - const auto &dense = state.peek(0); + const auto &vector = state.peek(0); auto m_cells = mixed.cells().typify<MCT>(); - auto v_cells = dense.cells().typify<VCT>(); + auto v_cells = vector.cells().typify<VCT>(); const auto &index = mixed.index(); size_t num_subspaces = index.size(); size_t num_output_cells = num_subspaces * param.out_subspace_size; ArrayRef<OCT> out_cells = state.stash.create_uninitialized_array<OCT>(num_output_cells); const MCT *m_cp = m_cells.begin(); const VCT *v_cp = v_cells.begin(); - OCT *out = out_cells.begin(); - for (size_t i = 0; i < num_output_cells; ++i) { - *out++ = my_dot_product(m_cp, v_cp, param.vector_size); + for (OCT &out : out_cells) { + out = my_dot_product(m_cp, v_cp, param.vector_size); m_cp += param.vector_size; } - assert(out == out_cells.end()); assert(m_cp == m_cells.end()); state.pop_pop_push(state.stash.create<ValueView>(param.res_type, index, TypedCells(out_cells))); } @@ -79,9 +77,9 @@ struct SelectMixedInnerProduct { } // namespace <unnamed> MixedInnerProductFunction::MixedInnerProductFunction(const ValueType &res_type_in, - const TensorFunction &lhs_in, - const TensorFunction &rhs_in) - : tensor_function::Op2(res_type_in, lhs_in, rhs_in) + const TensorFunction &mixed_child, + const TensorFunction &vector_child) + : tensor_function::Op2(res_type_in, mixed_child, vector_child) { } diff --git a/eval/src/vespa/eval/instruction/mixed_inner_product_function.h b/eval/src/vespa/eval/instruction/mixed_inner_product_function.h index 58a30d12adc..99ef88997b6 100644 --- a/eval/src/vespa/eval/instruction/mixed_inner_product_function.h +++ b/eval/src/vespa/eval/instruction/mixed_inner_product_function.h @@ -35,8 +35,8 @@ class MixedInnerProductFunction : public tensor_function::Op2 { public: MixedInnerProductFunction(const ValueType &res_type_in, - const TensorFunction &mix_in, - const TensorFunction &vec_in); + const TensorFunction &mixed_child, + const TensorFunction &vector_child); InterpretedFunction::Instruction compile_self(const ValueBuilderFactory &factory, Stash &stash) const override; bool result_is_mutable() const override { return true; } static bool compatible_types(const ValueType &res, const ValueType &mixed, const ValueType &dense); |