OmniSciDB  c0231cc57d
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
CompareIR.cpp
Go to the documentation of this file.
1 /*
2  * Copyright 2022 HEAVY.AI, Inc.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #include "CodeGenerator.h"
18 #include "Execute.h"
19 
20 #include <typeinfo>
21 
22 #include "../Parser/ParserNode.h"
23 
24 namespace {
25 
26 llvm::CmpInst::Predicate llvm_icmp_pred(const SQLOps op_type) {
27  switch (op_type) {
28  case kEQ:
29  return llvm::ICmpInst::ICMP_EQ;
30  case kNE:
31  return llvm::ICmpInst::ICMP_NE;
32  case kLT:
33  return llvm::ICmpInst::ICMP_SLT;
34  case kGT:
35  return llvm::ICmpInst::ICMP_SGT;
36  case kLE:
37  return llvm::ICmpInst::ICMP_SLE;
38  case kGE:
39  return llvm::ICmpInst::ICMP_SGE;
40  default:
41  abort();
42  }
43 }
44 
45 std::string icmp_name(const SQLOps op_type) {
46  switch (op_type) {
47  case kEQ:
48  return "eq";
49  case kNE:
50  return "ne";
51  case kLT:
52  return "lt";
53  case kGT:
54  return "gt";
55  case kLE:
56  return "le";
57  case kGE:
58  return "ge";
59  default:
60  abort();
61  }
62 }
63 
64 std::string icmp_arr_name(const SQLOps op_type) {
65  switch (op_type) {
66  case kEQ:
67  return "eq";
68  case kNE:
69  return "ne";
70  case kLT:
71  return "gt";
72  case kGT:
73  return "lt";
74  case kLE:
75  return "ge";
76  case kGE:
77  return "le";
78  default:
79  abort();
80  }
81 }
82 
83 llvm::CmpInst::Predicate llvm_fcmp_pred(const SQLOps op_type) {
84  switch (op_type) {
85  case kEQ:
86  return llvm::CmpInst::FCMP_OEQ;
87  case kNE:
88  return llvm::CmpInst::FCMP_ONE;
89  case kLT:
90  return llvm::CmpInst::FCMP_OLT;
91  case kGT:
92  return llvm::CmpInst::FCMP_OGT;
93  case kLE:
94  return llvm::CmpInst::FCMP_OLE;
95  case kGE:
96  return llvm::CmpInst::FCMP_OGE;
97  default:
98  abort();
99  }
100 }
101 
102 } // namespace
103 
104 namespace {
105 
106 std::string string_cmp_func(const SQLOps optype) {
107  switch (optype) {
108  case kLT:
109  return "string_lt";
110  case kLE:
111  return "string_le";
112  case kGT:
113  return "string_gt";
114  case kGE:
115  return "string_ge";
116  case kEQ:
117  return "string_eq";
118  case kNE:
119  return "string_ne";
120  default:
121  abort();
122  }
123 }
124 
125 std::shared_ptr<Analyzer::BinOper> lower_bw_eq(const Analyzer::BinOper* bw_eq) {
126  const auto eq_oper =
127  std::make_shared<Analyzer::BinOper>(bw_eq->get_type_info(),
128  bw_eq->get_contains_agg(),
129  kEQ,
130  bw_eq->get_qualifier(),
131  bw_eq->get_own_left_operand(),
132  bw_eq->get_own_right_operand());
133  const auto lhs_is_null =
134  std::make_shared<Analyzer::UOper>(kBOOLEAN, kISNULL, bw_eq->get_own_left_operand());
135  const auto rhs_is_null = std::make_shared<Analyzer::UOper>(
137  const auto both_are_null =
138  Parser::OperExpr::normalize(kAND, kONE, lhs_is_null, rhs_is_null);
139  const auto bw_eq_oper = std::dynamic_pointer_cast<Analyzer::BinOper>(
140  Parser::OperExpr::normalize(kOR, kONE, eq_oper, both_are_null));
141  CHECK(bw_eq_oper);
142  return bw_eq_oper;
143 }
144 
145 std::shared_ptr<Analyzer::BinOper> make_eq(const std::shared_ptr<Analyzer::Expr>& lhs,
146  const std::shared_ptr<Analyzer::Expr>& rhs,
147  const SQLOps optype) {
148  CHECK(IS_EQUIVALENCE(optype));
149  // Sides of a tuple equality are stripped of cast operators to simplify the logic
150  // in the hash table construction algorithm. Add them back here.
151  auto eq_oper = std::dynamic_pointer_cast<Analyzer::BinOper>(
152  Parser::OperExpr::normalize(optype, kONE, lhs, rhs));
153  CHECK(eq_oper);
154  return optype == kBW_EQ ? lower_bw_eq(eq_oper.get()) : eq_oper;
155 }
156 
157 // Convert a column tuple equality expression back to a conjunction of comparisons
158 // so that it can be handled by the regular code generation methods.
159 std::shared_ptr<Analyzer::BinOper> lower_multicol_compare(
160  const Analyzer::BinOper* multicol_compare) {
161  const auto left_tuple_expr = dynamic_cast<const Analyzer::ExpressionTuple*>(
162  multicol_compare->get_left_operand());
163  const auto right_tuple_expr = dynamic_cast<const Analyzer::ExpressionTuple*>(
164  multicol_compare->get_right_operand());
165  CHECK(left_tuple_expr && right_tuple_expr);
166  const auto& left_tuple = left_tuple_expr->getTuple();
167  const auto& right_tuple = right_tuple_expr->getTuple();
168  CHECK_EQ(left_tuple.size(), right_tuple.size());
169  CHECK_GT(left_tuple.size(), size_t(1));
170  auto acc =
171  make_eq(left_tuple.front(), right_tuple.front(), multicol_compare->get_optype());
172  for (size_t i = 1; i < left_tuple.size(); ++i) {
173  auto crt = make_eq(left_tuple[i], right_tuple[i], multicol_compare->get_optype());
174  const bool not_null =
175  acc->get_type_info().get_notnull() && crt->get_type_info().get_notnull();
176  acc = makeExpr<Analyzer::BinOper>(
177  SQLTypeInfo(kBOOLEAN, not_null), false, kAND, kONE, acc, crt);
178  }
179  return acc;
180 }
181 
183  auto lhs_cv = dynamic_cast<const Analyzer::ColumnVar*>(bin_oper->get_left_operand());
184  auto rhs_cv = dynamic_cast<const Analyzer::ColumnVar*>(bin_oper->get_right_operand());
185  auto comp_op = IS_COMPARISON(bin_oper->get_optype());
186  if (lhs_cv && rhs_cv && comp_op) {
187  auto lhs_ti = lhs_cv->get_type_info();
188  auto rhs_ti = rhs_cv->get_type_info();
189  if (lhs_ti.is_array() && rhs_ti.is_array()) {
190  throw std::runtime_error(
191  "Comparing two full array columns is not supported yet. Please consider "
192  "rewriting the full array comparison to a comparison between indexed array "
193  "columns "
194  "(i.e., arr1[1] {<, <=, >, >=} arr2[1]).");
195  }
196  }
197  auto lhs_bin_oper =
198  dynamic_cast<const Analyzer::BinOper*>(bin_oper->get_left_operand());
199  auto rhs_bin_oper =
200  dynamic_cast<const Analyzer::BinOper*>(bin_oper->get_right_operand());
201  // we can do (non-)equivalence check of two encoded string
202  // even if they are (indexed) array cols
203  auto theta_comp = IS_COMPARISON(bin_oper->get_optype()) &&
204  !IS_EQUIVALENCE(bin_oper->get_optype()) &&
205  bin_oper->get_optype() != SQLOps::kNE;
206  if (lhs_bin_oper && rhs_bin_oper && theta_comp &&
207  lhs_bin_oper->get_optype() == SQLOps::kARRAY_AT &&
208  rhs_bin_oper->get_optype() == SQLOps::kARRAY_AT) {
209  auto lhs_arr_cv =
210  dynamic_cast<const Analyzer::ColumnVar*>(lhs_bin_oper->get_left_operand());
211  auto lhs_arr_idx =
212  dynamic_cast<const Analyzer::Constant*>(lhs_bin_oper->get_right_operand());
213  auto rhs_arr_cv =
214  dynamic_cast<const Analyzer::ColumnVar*>(rhs_bin_oper->get_left_operand());
215  auto rhs_arr_idx =
216  dynamic_cast<const Analyzer::Constant*>(rhs_bin_oper->get_right_operand());
217  if (lhs_arr_cv && rhs_arr_cv && lhs_arr_idx && rhs_arr_idx &&
218  ((lhs_arr_cv->get_type_info().is_array() &&
219  lhs_arr_cv->get_type_info().get_subtype() == SQLTypes::kTEXT) ||
220  (rhs_arr_cv->get_type_info().is_string() &&
221  rhs_arr_cv->get_type_info().get_subtype() == SQLTypes::kTEXT))) {
222  throw std::runtime_error(
223  "Comparison between string array columns is not supported yet.");
224  }
225  }
226 }
227 
228 } // namespace
229 
230 llvm::Value* CodeGenerator::codegenCmp(const Analyzer::BinOper* bin_oper,
231  const CompilationOptions& co) {
233  const auto qualifier = bin_oper->get_qualifier();
234  const auto lhs = bin_oper->get_left_operand();
235  const auto rhs = bin_oper->get_right_operand();
236  if (dynamic_cast<const Analyzer::ExpressionTuple*>(lhs)) {
237  CHECK(dynamic_cast<const Analyzer::ExpressionTuple*>(rhs));
238  const auto lowered = lower_multicol_compare(bin_oper);
239  const auto lowered_lvs = codegen(lowered.get(), true, co);
240  CHECK_EQ(size_t(1), lowered_lvs.size());
241  return lowered_lvs.front();
242  }
243  const auto optype = bin_oper->get_optype();
244  if (optype == kBW_EQ) {
245  const auto bw_eq_oper = lower_bw_eq(bin_oper);
246  return codegenLogical(bw_eq_oper.get(), co);
247  }
248  if (optype == kOVERLAPS) {
249  return codegenOverlaps(optype,
250  qualifier,
251  bin_oper->get_own_left_operand(),
252  bin_oper->get_own_right_operand(),
253  co);
254  }
255  if (is_unnest(lhs) || is_unnest(rhs)) {
256  throw std::runtime_error("Unnest not supported in comparisons");
257  }
258  check_array_comp_cond(bin_oper);
259  const auto& lhs_ti = lhs->get_type_info();
260  const auto& rhs_ti = rhs->get_type_info();
261 
262  if (lhs_ti.is_string() && rhs_ti.is_string() &&
263  !(IS_EQUIVALENCE(optype) || optype == kNE)) {
264  auto cmp_str = codegenStrCmp(optype,
265  qualifier,
266  bin_oper->get_own_left_operand(),
267  bin_oper->get_own_right_operand(),
268  co);
269  if (cmp_str) {
270  return cmp_str;
271  }
272  }
273 
274  if (lhs_ti.is_decimal()) {
275  auto cmp_decimal_const =
276  codegenCmpDecimalConst(optype, qualifier, lhs, lhs_ti, rhs, co);
277  if (cmp_decimal_const) {
278  return cmp_decimal_const;
279  }
280  }
281  auto lhs_lvs = codegen(lhs, true, co);
282  return codegenCmp(optype, qualifier, lhs_lvs, lhs_ti, rhs, co);
283 }
284 
285 llvm::Value* CodeGenerator::codegenOverlaps(const SQLOps optype,
286  const SQLQualifier qualifier,
287  const std::shared_ptr<Analyzer::Expr> lhs,
288  const std::shared_ptr<Analyzer::Expr> rhs,
289  const CompilationOptions& co) {
291  const auto lhs_ti = lhs->get_type_info();
293  // failed to build a suitable hash table. short circuit the overlaps expression by
294  // always returning true. this will fall into the ST_Contains check, which will do
295  // overlaps checks before the heavier contains computation.
296  VLOG(1) << "Failed to build overlaps hash table, short circuiting overlaps operator.";
297  return llvm::ConstantInt::get(get_int_type(8, cgen_state_->context_), true);
298  }
299  // TODO(adb): we should never get here, but going to leave this in place for now since
300  // it will likely be useful in factoring the bounds check out of ST_Contains
301  CHECK(lhs_ti.is_geometry());
302 
303  if (lhs_ti.is_geometry()) {
304  // only point in linestring/poly/mpoly is currently supported
305  CHECK(lhs_ti.get_type() == kPOINT);
306  const auto lhs_col = dynamic_cast<Analyzer::ColumnVar*>(lhs.get());
307  CHECK(lhs_col);
308 
309  // Get the actual point data column descriptor
310  const auto coords_cd = executor()->getCatalog()->getMetadataForColumn(
311  lhs_col->get_table_id(), lhs_col->get_column_id() + 1);
312  CHECK(coords_cd);
313 
314  std::vector<std::shared_ptr<Analyzer::Expr>> geoargs;
315  geoargs.push_back(makeExpr<Analyzer::ColumnVar>(coords_cd->columnType,
316  coords_cd->tableId,
317  coords_cd->columnId,
318  lhs_col->get_rte_idx()));
319 
320  Datum input_compression;
321  input_compression.intval =
322  (lhs_ti.get_compression() == kENCODING_GEOINT && lhs_ti.get_comp_param() == 32)
323  ? 1
324  : 0;
325  geoargs.push_back(makeExpr<Analyzer::Constant>(kINT, false, input_compression));
326  Datum input_srid;
327  input_srid.intval = lhs_ti.get_input_srid();
328  geoargs.push_back(makeExpr<Analyzer::Constant>(kINT, false, input_srid));
329  Datum output_srid;
330  output_srid.intval = lhs_ti.get_output_srid();
331  geoargs.push_back(makeExpr<Analyzer::Constant>(kINT, false, output_srid));
332 
333  const auto x_ptr_oper = makeExpr<Analyzer::FunctionOper>(
334  SQLTypeInfo(kDOUBLE, true), "ST_X_Point", geoargs);
335  const auto y_ptr_oper = makeExpr<Analyzer::FunctionOper>(
336  SQLTypeInfo(kDOUBLE, true), "ST_Y_Point", geoargs);
337 
338  const auto rhs_ti = rhs->get_type_info();
339  CHECK(IS_GEO_POLY(rhs_ti.get_type()));
340  const auto rhs_col = dynamic_cast<Analyzer::ColumnVar*>(rhs.get());
341  CHECK(rhs_col);
342 
343  const auto poly_bounds_cd = executor()->getCatalog()->getMetadataForColumn(
344  rhs_col->get_table_id(),
345  rhs_col->get_column_id() + rhs_ti.get_physical_coord_cols() + 1);
346  CHECK(poly_bounds_cd);
347 
348  auto bbox_col_var = makeExpr<Analyzer::ColumnVar>(poly_bounds_cd->columnType,
349  poly_bounds_cd->tableId,
350  poly_bounds_cd->columnId,
351  rhs_col->get_rte_idx());
352 
353  const auto bbox_contains_func_oper =
354  makeExpr<Analyzer::FunctionOper>(SQLTypeInfo(kBOOLEAN, false),
355  "Point_Overlaps_Box",
356  std::vector<std::shared_ptr<Analyzer::Expr>>{
357  bbox_col_var, x_ptr_oper, y_ptr_oper});
358 
359  return codegenFunctionOper(bbox_contains_func_oper.get(), co);
360  }
361 
362  CHECK(false) << "Unsupported type for overlaps operator: " << lhs_ti.get_type_name();
363  return nullptr;
364 }
365 
366 llvm::Value* CodeGenerator::codegenStrCmp(const SQLOps optype,
367  const SQLQualifier qualifier,
368  const std::shared_ptr<Analyzer::Expr> lhs,
369  const std::shared_ptr<Analyzer::Expr> rhs,
370  const CompilationOptions& co) {
372  const auto lhs_ti = lhs->get_type_info();
373  const auto rhs_ti = rhs->get_type_info();
374 
375  CHECK(lhs_ti.is_string());
376  CHECK(rhs_ti.is_string());
377 
378  const auto null_check_suffix = get_null_check_suffix(lhs_ti, rhs_ti);
379  if (lhs_ti.get_compression() == kENCODING_DICT &&
380  rhs_ti.get_compression() == kENCODING_DICT) {
381  if (lhs_ti.get_comp_param() == rhs_ti.get_comp_param()) {
382  // Both operands share a dictionary
383 
384  // check if query is trying to compare a columnt against literal
385 
386  auto ir = codegenDictStrCmp(lhs, rhs, optype, co);
387  if (ir) {
388  return ir;
389  }
390  } else {
391  // Both operands don't share a dictionary
392  return nullptr;
393  }
394  }
395  return nullptr;
396 }
397 
399  const SQLQualifier qualifier,
400  const Analyzer::Expr* lhs,
401  const SQLTypeInfo& lhs_ti,
402  const Analyzer::Expr* rhs,
403  const CompilationOptions& co) {
405  auto u_oper = dynamic_cast<const Analyzer::UOper*>(lhs);
406  if (!u_oper || u_oper->get_optype() != kCAST) {
407  return nullptr;
408  }
409  auto rhs_constant = dynamic_cast<const Analyzer::Constant*>(rhs);
410  if (!rhs_constant) {
411  return nullptr;
412  }
413  const auto operand = u_oper->get_operand();
414  const auto& operand_ti = operand->get_type_info();
415  if (operand_ti.is_decimal() && operand_ti.get_scale() < lhs_ti.get_scale()) {
416  // lhs decimal type has smaller scale
417  } else if (operand_ti.is_integer() && 0 < lhs_ti.get_scale()) {
418  // lhs is integer, no need to scale it all the way up to the cmp expr scale
419  } else {
420  return nullptr;
421  }
422 
423  auto scale_diff = lhs_ti.get_scale() - operand_ti.get_scale() - 1;
424  int64_t bigintval = rhs_constant->get_constval().bigintval;
425  bool negative = false;
426  if (bigintval < 0) {
427  negative = true;
428  bigintval = -bigintval;
429  }
430  int64_t truncated_decimal = bigintval / exp_to_scale(scale_diff);
431  int64_t decimal_tail = bigintval % exp_to_scale(scale_diff);
432  if (truncated_decimal % 10 == 0 && decimal_tail > 0) {
433  truncated_decimal += 1;
434  }
437  lhs_ti.get_scale() - scale_diff,
438  operand_ti.get_notnull());
439  if (negative) {
440  truncated_decimal = -truncated_decimal;
441  }
442  Datum d;
443  d.bigintval = truncated_decimal;
444  const auto new_rhs_lit =
445  makeExpr<Analyzer::Constant>(new_ti, rhs_constant->get_is_null(), d);
446  const auto operand_lv = codegen(operand, true, co).front();
447  const auto lhs_lv = codegenCast(operand_lv, operand_ti, new_ti, false, co);
448  return codegenCmp(optype, qualifier, {lhs_lv}, new_ti, new_rhs_lit.get(), co);
449 }
450 
451 llvm::Value* CodeGenerator::codegenCmp(const SQLOps optype,
452  const SQLQualifier qualifier,
453  std::vector<llvm::Value*> lhs_lvs,
454  const SQLTypeInfo& lhs_ti,
455  const Analyzer::Expr* rhs,
456  const CompilationOptions& co) {
458  CHECK(IS_COMPARISON(optype));
459  const auto& rhs_ti = rhs->get_type_info();
460  if (rhs_ti.is_array()) {
461  return codegenQualifierCmp(optype, qualifier, lhs_lvs, rhs, co);
462  }
463  auto rhs_lvs = codegen(rhs, true, co);
464  CHECK_EQ(kONE, qualifier);
465  if (optype == kOVERLAPS) {
466  CHECK(lhs_ti.is_geometry());
467  CHECK(rhs_ti.is_array() ||
468  rhs_ti.is_geometry()); // allow geo col or bounds col to pass
469  } else {
470  CHECK((lhs_ti.get_type() == rhs_ti.get_type()) ||
471  (lhs_ti.is_string() && rhs_ti.is_string()));
472  }
473  const auto null_check_suffix = get_null_check_suffix(lhs_ti, rhs_ti);
474  if (lhs_ti.is_integer() || lhs_ti.is_decimal() || lhs_ti.is_time() ||
475  lhs_ti.is_boolean() || lhs_ti.is_string() || lhs_ti.is_timeinterval()) {
476  if (lhs_ti.is_string()) {
477  CHECK(rhs_ti.is_string());
478  CHECK_EQ(lhs_ti.get_compression(), rhs_ti.get_compression());
479  if (lhs_ti.get_compression() == kENCODING_NONE) {
480  // unpack pointer + length if necessary
481  if (lhs_lvs.size() != 3) {
482  CHECK_EQ(size_t(1), lhs_lvs.size());
483  lhs_lvs.push_back(cgen_state_->emitCall("extract_str_ptr", {lhs_lvs.front()}));
484  lhs_lvs.push_back(cgen_state_->emitCall("extract_str_len", {lhs_lvs.front()}));
485  }
486  if (rhs_lvs.size() != 3) {
487  CHECK_EQ(size_t(1), rhs_lvs.size());
488  rhs_lvs.push_back(cgen_state_->emitCall("extract_str_ptr", {rhs_lvs.front()}));
489  rhs_lvs.push_back(cgen_state_->emitCall("extract_str_len", {rhs_lvs.front()}));
490  }
491  std::vector<llvm::Value*> str_cmp_args{
492  lhs_lvs[1], lhs_lvs[2], rhs_lvs[1], rhs_lvs[2]};
493  if (!null_check_suffix.empty()) {
494  str_cmp_args.push_back(
495  cgen_state_->inlineIntNull(SQLTypeInfo(kBOOLEAN, false)));
496  }
497  return cgen_state_->emitCall(
498  string_cmp_func(optype) + (null_check_suffix.empty() ? "" : "_nullable"),
499  str_cmp_args);
500  } else {
501  CHECK(optype == kEQ || optype == kNE);
502  }
503  }
504 
505  if (lhs_ti.is_boolean() && rhs_ti.is_boolean()) {
506  auto& lhs_lv = lhs_lvs.front();
507  auto& rhs_lv = rhs_lvs.front();
508  CHECK(lhs_lv->getType()->isIntegerTy());
509  CHECK(rhs_lv->getType()->isIntegerTy());
510  if (lhs_lv->getType()->getIntegerBitWidth() <
511  rhs_lv->getType()->getIntegerBitWidth()) {
512  lhs_lv =
513  cgen_state_->castToTypeIn(lhs_lv, rhs_lv->getType()->getIntegerBitWidth());
514  } else {
515  rhs_lv =
516  cgen_state_->castToTypeIn(rhs_lv, lhs_lv->getType()->getIntegerBitWidth());
517  }
518  }
519 
520  return null_check_suffix.empty()
521  ? cgen_state_->ir_builder_.CreateICmp(
522  llvm_icmp_pred(optype), lhs_lvs.front(), rhs_lvs.front())
523  : cgen_state_->emitCall(
524  icmp_name(optype) + "_" + numeric_type_name(lhs_ti) +
525  null_check_suffix,
526  {lhs_lvs.front(),
527  rhs_lvs.front(),
528  cgen_state_->llInt(inline_int_null_val(lhs_ti)),
529  cgen_state_->inlineIntNull(SQLTypeInfo(kBOOLEAN, false))});
530  }
531  if (lhs_ti.get_type() == kFLOAT || lhs_ti.get_type() == kDOUBLE) {
532  return null_check_suffix.empty()
533  ? cgen_state_->ir_builder_.CreateFCmp(
534  llvm_fcmp_pred(optype), lhs_lvs.front(), rhs_lvs.front())
535  : cgen_state_->emitCall(
536  icmp_name(optype) + "_" + numeric_type_name(lhs_ti) +
537  null_check_suffix,
538  {lhs_lvs.front(),
539  rhs_lvs.front(),
540  lhs_ti.get_type() == kFLOAT ? cgen_state_->llFp(NULL_FLOAT)
541  : cgen_state_->llFp(NULL_DOUBLE),
542  cgen_state_->inlineIntNull(SQLTypeInfo(kBOOLEAN, false))});
543  }
544  CHECK(false);
545  return nullptr;
546 }
547 
548 llvm::Value* CodeGenerator::codegenQualifierCmp(const SQLOps optype,
549  const SQLQualifier qualifier,
550  std::vector<llvm::Value*> lhs_lvs,
551  const Analyzer::Expr* rhs,
552  const CompilationOptions& co) {
554  const auto& rhs_ti = rhs->get_type_info();
555  const Analyzer::Expr* arr_expr{rhs};
556  if (dynamic_cast<const Analyzer::UOper*>(rhs)) {
557  const auto cast_arr = static_cast<const Analyzer::UOper*>(rhs);
558  CHECK_EQ(kCAST, cast_arr->get_optype());
559  arr_expr = cast_arr->get_operand();
560  }
561  const auto& arr_ti = arr_expr->get_type_info();
562  const auto& elem_ti = arr_ti.get_elem_type();
563  auto rhs_lvs = codegen(arr_expr, true, co);
564  CHECK_NE(kONE, qualifier);
565  std::string fname{std::string("array_") + (qualifier == kANY ? "any" : "all") + "_" +
566  icmp_arr_name(optype)};
567  const auto& target_ti = rhs_ti.get_elem_type();
568  const bool is_real_string{target_ti.is_string() &&
569  target_ti.get_compression() != kENCODING_DICT};
570  if (is_real_string) {
571  if (g_cluster) {
572  throw std::runtime_error(
573  "Comparison between a dictionary-encoded and a none-encoded string not "
574  "supported for distributed queries");
575  }
576  if (g_enable_watchdog) {
577  throw WatchdogException(
578  "Comparison between a dictionary-encoded and a none-encoded string would be "
579  "slow");
580  }
582  throw QueryMustRunOnCpu();
583  }
584  CHECK_EQ(kENCODING_NONE, target_ti.get_compression());
585  fname += "_str";
586  }
587  if (elem_ti.is_integer() || elem_ti.is_boolean() || elem_ti.is_string() ||
588  elem_ti.is_decimal()) {
589  fname += ("_" + numeric_type_name(elem_ti));
590  } else {
591  CHECK(elem_ti.is_fp());
592  fname += elem_ti.get_type() == kDOUBLE ? "_double" : "_float";
593  }
594  if (is_real_string) {
595  CHECK_EQ(size_t(3), lhs_lvs.size());
597  fname,
599  {rhs_lvs.front(),
600  posArg(arr_expr),
601  lhs_lvs[1],
602  lhs_lvs[2],
603  cgen_state_->llInt(int64_t(executor()->getStringDictionaryProxy(
604  elem_ti.get_comp_param(), executor()->getRowSetMemoryOwner(), true))),
605  cgen_state_->inlineIntNull(elem_ti)});
606  }
607  if (target_ti.is_integer() || target_ti.is_boolean() || target_ti.is_string() ||
608  target_ti.is_decimal()) {
609  fname += ("_" + numeric_type_name(target_ti));
610  } else {
611  CHECK(target_ti.is_fp());
612  fname += target_ti.get_type() == kDOUBLE ? "_double" : "_float";
613  }
614  return cgen_state_->emitExternalCall(
615  fname,
616  get_int_type(1, cgen_state_->context_),
617  {rhs_lvs.front(),
618  posArg(arr_expr),
619  lhs_lvs.front(),
620  elem_ti.is_fp() ? static_cast<llvm::Value*>(cgen_state_->inlineFpNull(elem_ti))
621  : static_cast<llvm::Value*>(cgen_state_->inlineIntNull(elem_ti))});
622 }
void check_array_comp_cond(const Analyzer::BinOper *bin_oper)
Definition: CompareIR.cpp:182
static constexpr int32_t kMaxRepresentableNumericPrecision
Definition: sqltypes.h:47
#define CHECK_EQ(x, y)
Definition: Logger.h:230
#define NULL_DOUBLE
llvm::Value * codegenStrCmp(const SQLOps, const SQLQualifier, const std::shared_ptr< Analyzer::Expr >, const std::shared_ptr< Analyzer::Expr >, const CompilationOptions &)
Definition: CompareIR.cpp:366
std::string icmp_name(const SQLOps op_type)
Definition: CompareIR.cpp:45
#define IS_EQUIVALENCE(X)
Definition: sqldefs.h:68
CgenState * cgen_state_
#define NULL_FLOAT
SQLQualifier
Definition: sqldefs.h:70
std::shared_ptr< Analyzer::BinOper > lower_multicol_compare(const Analyzer::BinOper *multicol_compare)
Definition: CompareIR.cpp:159
HOST DEVICE int get_scale() const
Definition: sqltypes.h:409
llvm::Value * emitExternalCall(const std::string &fname, llvm::Type *ret_type, const std::vector< llvm::Value * > args, const std::vector< llvm::Attribute::AttrKind > &fnattrs={}, const bool has_struct_return=false)
Definition: CgenState.h:217
const Expr * get_right_operand() const
Definition: Analyzer.h:451
SQLOps
Definition: sqldefs.h:28
Definition: sqldefs.h:34
static std::shared_ptr< Analyzer::Expr > normalize(const SQLOps optype, const SQLQualifier qual, std::shared_ptr< Analyzer::Expr > left_expr, std::shared_ptr< Analyzer::Expr > right_expr, const Executor *executor=nullptr)
Definition: ParserNode.cpp:370
Definition: sqldefs.h:35
llvm::Value * posArg(const Analyzer::Expr *) const
Definition: ColumnIR.cpp:550
Definition: sqldefs.h:37
bool get_contains_agg() const
Definition: Analyzer.h:84
Definition: sqldefs.h:48
Definition: sqldefs.h:29
std::shared_ptr< Analyzer::BinOper > lower_bw_eq(const Analyzer::BinOper *bw_eq)
Definition: CompareIR.cpp:125
HOST DEVICE SQLTypes get_type() const
Definition: sqltypes.h:404
llvm::CmpInst::Predicate llvm_fcmp_pred(const SQLOps op_type)
Definition: CompareIR.cpp:83
llvm::Type * get_int_type(const int width, llvm::LLVMContext &context)
#define CHECK_GT(x, y)
Definition: Logger.h:234
int32_t intval
Definition: sqltypes.h:234
bool is_time() const
Definition: sqltypes.h:606
std::string icmp_arr_name(const SQLOps op_type)
Definition: CompareIR.cpp:64
bool g_enable_overlaps_hashjoin
Definition: Execute.cpp:102
SQLOps get_optype() const
Definition: Analyzer.h:447
llvm::LLVMContext & context_
Definition: CgenState.h:439
llvm::Value * codegenCmpDecimalConst(const SQLOps, const SQLQualifier, const Analyzer::Expr *, const SQLTypeInfo &, const Analyzer::Expr *, const CompilationOptions &)
Definition: CompareIR.cpp:398
bool is_integer() const
Definition: sqltypes.h:602
#define CHECK_NE(x, y)
Definition: Logger.h:231
llvm::ConstantInt * inlineIntNull(const SQLTypeInfo &)
Definition: CgenState.cpp:64
int64_t bigintval
Definition: sqltypes.h:235
bool is_timeinterval() const
Definition: sqltypes.h:611
llvm::Value * codegenFunctionOper(const Analyzer::FunctionOper *, const CompilationOptions &)
Definition: sqldefs.h:36
bool g_enable_watchdog
Definition: sqldefs.h:70
llvm::Value * codegenOverlaps(const SQLOps, const SQLQualifier, const std::shared_ptr< Analyzer::Expr >, const std::shared_ptr< Analyzer::Expr >, const CompilationOptions &)
Definition: CompareIR.cpp:285
llvm::Value * codegenDictStrCmp(const std::shared_ptr< Analyzer::Expr >, const std::shared_ptr< Analyzer::Expr >, const SQLOps, const CompilationOptions &co)
bool is_boolean() const
Definition: sqltypes.h:607
#define AUTOMATIC_IR_METADATA(CGENSTATE)
std::string get_null_check_suffix(const SQLTypeInfo &lhs_ti, const SQLTypeInfo &rhs_ti)
Definition: Execute.h:1438
const SQLTypeInfo & get_type_info() const
Definition: Analyzer.h:82
llvm::Value * emitCall(const std::string &fname, const std::vector< llvm::Value * > &args)
Definition: CgenState.cpp:219
ExecutorDeviceType device_type
Definition: sqldefs.h:33
std::vector< llvm::Value * > codegen(const Analyzer::Expr *, const bool fetch_columns, const CompilationOptions &)
Definition: IRCodegen.cpp:30
Definition: sqltypes.h:66
Definition: sqldefs.h:70
HOST DEVICE EncodingType get_compression() const
Definition: sqltypes.h:412
std::shared_ptr< Analyzer::BinOper > make_eq(const std::shared_ptr< Analyzer::Expr > &lhs, const std::shared_ptr< Analyzer::Expr > &rhs, const SQLOps optype)
Definition: CompareIR.cpp:145
llvm::Value * codegenQualifierCmp(const SQLOps, const SQLQualifier, std::vector< llvm::Value * >, const Analyzer::Expr *, const CompilationOptions &)
Definition: CompareIR.cpp:548
Definition: sqldefs.h:31
llvm::CmpInst::Predicate llvm_icmp_pred(const SQLOps op_type)
Definition: CompareIR.cpp:26
llvm::Value * codegenCmp(const Analyzer::BinOper *, const CompilationOptions &)
Definition: CompareIR.cpp:230
llvm::ConstantInt * llInt(const T v) const
Definition: CgenState.h:306
#define CHECK(condition)
Definition: Logger.h:222
bool is_geometry() const
Definition: sqltypes.h:612
Definition: sqldefs.h:30
llvm::Value * codegenLogical(const Analyzer::BinOper *, const CompilationOptions &)
Definition: LogicalIR.cpp:298
uint64_t exp_to_scale(const unsigned exp)
int64_t inline_int_null_val(const SQL_TYPE_INFO &ti)
bool g_cluster
Definition: sqldefs.h:32
const Expr * get_left_operand() const
Definition: Analyzer.h:450
llvm::Value * codegenCast(const Analyzer::UOper *, const CompilationOptions &)
Definition: CastIR.cpp:21
std::string numeric_type_name(const SQLTypeInfo &ti)
Definition: Execute.h:209
Definition: sqltypes.h:59
bool is_unnest(const Analyzer::Expr *expr)
Definition: Execute.h:1454
bool is_string() const
Definition: sqltypes.h:600
const std::shared_ptr< Analyzer::Expr > get_own_right_operand() const
Definition: Analyzer.h:455
bool is_decimal() const
Definition: sqltypes.h:603
const std::shared_ptr< Analyzer::Expr > get_own_left_operand() const
Definition: Analyzer.h:452
#define VLOG(n)
Definition: Logger.h:316
std::string string_cmp_func(const SQLOps optype)
Definition: CompareIR.cpp:106
#define IS_COMPARISON(X)
Definition: sqldefs.h:57
SQLQualifier get_qualifier() const
Definition: Analyzer.h:449
Executor * executor() const
#define IS_GEO_POLY(T)
Definition: sqltypes.h:328