hcq/oceanbase
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[FEAT MERGE] impl vectorization 2.0

Browse Source 
Co-authored-by: Naynahs <cfzy002@126.com>
Co-authored-by: hwx65 <1780011298@qq.com>
Co-authored-by: oceanoverflow <oceanoverflow@gmail.com>
This commit is contained in:
obdev
2023-12-22 03:43:19 +00:00
committed by ob-robot
parent 1178245448
commit b6773084c6
592 changed files with 358124 additions and 303288 deletions
Download Patch File Download Diff File Expand all files Collapse all files

93
src/sql/engine/expr/ob_expr_sin.cpp
View File

@ -52,6 +52,99 @@ int ObExprSin::cg_expr(ObExprCGCtx &expr_cg_ctx, const ObRawExpr &raw_expr,
UNUSED(expr_cg_ctx);
UNUSED(raw_expr);
rt_expr.eval_func_ = calc_sin_expr;
if (ObDoubleType == rt_expr.args_[0]->datum_meta_.type_) {
rt_expr.eval_vector_func_ = eval_double_sin_vector;
} else if (ObNumberType == rt_expr.args_[0]->datum_meta_.type_) {
rt_expr.eval_vector_func_ = eval_number_sin_vector;
} else {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("invalid arg type", K(rt_expr.args_[0]->datum_meta_.type_), K(ret));
}
return ret;
}
template <typename ArgVec, typename ResVec, bool IS_DOUBLE>
static int vector_sin(const ObExpr &expr,
ObEvalCtx &ctx,
const ObBitVector &skip,
const EvalBound &bound)
{
int ret = OB_SUCCESS;
ArgVec *arg_vec = static_cast<ArgVec *>(expr.args_[0]->get_vector(ctx));
ResVec *res_vec = static_cast<ResVec *>(expr.get_vector(ctx));
ObBitVector &eval_flags = expr.get_evaluated_flags(ctx);
for (int64_t idx = bound.start(); OB_SUCC(ret) && idx < bound.end(); ++idx) {
if (skip.at(idx) || eval_flags.at(idx)) {
continue;
} else if (arg_vec->is_null(idx)) {
res_vec->set_null(idx);
} else if (IS_DOUBLE) {
const double arg = arg_vec->get_double(idx);
double res = sin(arg);
res_vec->set_double(idx, res);
} else {
number::ObNumber res_nmb;
number::ObNumber radian_nmb(arg_vec->get_number(idx));
ObEvalCtx::TempAllocGuard alloc_guard(ctx);
if (OB_FAIL(radian_nmb.sin(res_nmb, alloc_guard.get_allocator()))) {
LOG_WARN("calc expr failed", K(ret), K(radian_nmb), K(expr));
} else {
res_vec->set_number(idx, res_nmb);
}
}
eval_flags.set(idx);
}
return ret;
}
int ObExprSin::eval_number_sin_vector(const ObExpr &expr,
ObEvalCtx &ctx,
const ObBitVector &skip,
const EvalBound &bound)
{
int ret = OB_SUCCESS;
if (OB_FAIL(expr.args_[0]->eval_vector(ctx, skip, bound))) {
LOG_WARN("fail to eval sin param", K(ret));
} else {
VectorFormat arg_format = expr.args_[0]->get_format(ctx);
VectorFormat res_format = expr.get_format(ctx);
if (VEC_DISCRETE == arg_format && VEC_DISCRETE == res_format) {
ret = vector_sin<NumberDiscVec, NumberDiscVec, false>(expr, ctx, skip, bound);
} else if (VEC_UNIFORM == arg_format && VEC_DISCRETE == res_format) {
ret = vector_sin<NumberUniVec, NumberDiscVec, false>(expr, ctx, skip, bound);
} else if (VEC_CONTINUOUS == arg_format && VEC_DISCRETE == res_format) {
ret = vector_sin<NumberContVec, NumberDiscVec, false>(expr, ctx, skip, bound);
//...
} else {
ret = vector_sin<ObVectorBase, ObVectorBase, false>(expr, ctx, skip, bound);
}
}
return ret;
}
int ObExprSin::eval_double_sin_vector(const ObExpr &expr,
ObEvalCtx &ctx,
const ObBitVector &skip,
const EvalBound &bound)
{
int ret = OB_SUCCESS;
if (OB_FAIL(expr.args_[0]->eval_vector(ctx, skip, bound))) {
LOG_WARN("fail to eval sin param", K(ret));
} else {
VectorFormat arg_format = expr.args_[0]->get_format(ctx);
VectorFormat res_format = expr.get_format(ctx);
if (VEC_FIXED == arg_format && VEC_FIXED == res_format) {
vector_sin<DoubleFixedVec, DoubleFixedVec, true>(expr, ctx, skip, bound);
} else if (VEC_UNIFORM == arg_format && VEC_FIXED == res_format) {
vector_sin<DoubleUniVec, DoubleFixedVec, true>(expr, ctx, skip, bound);
//...
} else {
vector_sin<ObVectorBase, ObVectorBase, true>(expr, ctx, skip, bound);
}
}
return ret;
}