OmniSciDB  fe05a0c208
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
CompareIR.cpp
Go to the documentation of this file.
1 /*
2  * Copyright 2017 MapD Technologies, 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  }
435  SQLTypeInfo new_ti = SQLTypeInfo(
436  kDECIMAL, 19, lhs_ti.get_scale() - scale_diff, operand_ti.get_notnull());
437  if (negative) {
438  truncated_decimal = -truncated_decimal;
439  }
440  Datum d;
441  d.bigintval = truncated_decimal;
442  const auto new_rhs_lit =
443  makeExpr<Analyzer::Constant>(new_ti, rhs_constant->get_is_null(), d);
444  const auto operand_lv = codegen(operand, true, co).front();
445  const auto lhs_lv = codegenCast(operand_lv, operand_ti, new_ti, false, co);
446  return codegenCmp(optype, qualifier, {lhs_lv}, new_ti, new_rhs_lit.get(), co);
447 }
448 
449 llvm::Value* CodeGenerator::codegenCmp(const SQLOps optype,
450  const SQLQualifier qualifier,
451  std::vector<llvm::Value*> lhs_lvs,
452  const SQLTypeInfo& lhs_ti,
453  const Analyzer::Expr* rhs,
454  const CompilationOptions& co) {
456  CHECK(IS_COMPARISON(optype));
457  const auto& rhs_ti = rhs->get_type_info();
458  if (rhs_ti.is_array()) {
459  return codegenQualifierCmp(optype, qualifier, lhs_lvs, rhs, co);
460  }
461  auto rhs_lvs = codegen(rhs, true, co);
462  CHECK_EQ(kONE, qualifier);
463  if (optype == kOVERLAPS) {
464  CHECK(lhs_ti.is_geometry());
465  CHECK(rhs_ti.is_array() ||
466  rhs_ti.is_geometry()); // allow geo col or bounds col to pass
467  } else {
468  CHECK((lhs_ti.get_type() == rhs_ti.get_type()) ||
469  (lhs_ti.is_string() && rhs_ti.is_string()));
470  }
471  const auto null_check_suffix = get_null_check_suffix(lhs_ti, rhs_ti);
472  if (lhs_ti.is_integer() || lhs_ti.is_decimal() || lhs_ti.is_time() ||
473  lhs_ti.is_boolean() || lhs_ti.is_string() || lhs_ti.is_timeinterval()) {
474  if (lhs_ti.is_string()) {
475  CHECK(rhs_ti.is_string());
476  CHECK_EQ(lhs_ti.get_compression(), rhs_ti.get_compression());
477  if (lhs_ti.get_compression() == kENCODING_NONE) {
478  // unpack pointer + length if necessary
479  if (lhs_lvs.size() != 3) {
480  CHECK_EQ(size_t(1), lhs_lvs.size());
481  lhs_lvs.push_back(cgen_state_->emitCall("extract_str_ptr", {lhs_lvs.front()}));
482  lhs_lvs.push_back(cgen_state_->emitCall("extract_str_len", {lhs_lvs.front()}));
483  }
484  if (rhs_lvs.size() != 3) {
485  CHECK_EQ(size_t(1), rhs_lvs.size());
486  rhs_lvs.push_back(cgen_state_->emitCall("extract_str_ptr", {rhs_lvs.front()}));
487  rhs_lvs.push_back(cgen_state_->emitCall("extract_str_len", {rhs_lvs.front()}));
488  }
489  std::vector<llvm::Value*> str_cmp_args{
490  lhs_lvs[1], lhs_lvs[2], rhs_lvs[1], rhs_lvs[2]};
491  if (!null_check_suffix.empty()) {
492  str_cmp_args.push_back(
493  cgen_state_->inlineIntNull(SQLTypeInfo(kBOOLEAN, false)));
494  }
495  return cgen_state_->emitCall(
496  string_cmp_func(optype) + (null_check_suffix.empty() ? "" : "_nullable"),
497  str_cmp_args);
498  } else {
499  CHECK(optype == kEQ || optype == kNE);
500  }
501  }
502 
503  if (lhs_ti.is_boolean() && rhs_ti.is_boolean()) {
504  auto& lhs_lv = lhs_lvs.front();
505  auto& rhs_lv = rhs_lvs.front();
506  CHECK(lhs_lv->getType()->isIntegerTy());
507  CHECK(rhs_lv->getType()->isIntegerTy());
508  if (lhs_lv->getType()->getIntegerBitWidth() <
509  rhs_lv->getType()->getIntegerBitWidth()) {
510  lhs_lv =
511  cgen_state_->castToTypeIn(lhs_lv, rhs_lv->getType()->getIntegerBitWidth());
512  } else {
513  rhs_lv =
514  cgen_state_->castToTypeIn(rhs_lv, lhs_lv->getType()->getIntegerBitWidth());
515  }
516  }
517 
518  return null_check_suffix.empty()
519  ? cgen_state_->ir_builder_.CreateICmp(
520  llvm_icmp_pred(optype), lhs_lvs.front(), rhs_lvs.front())
521  : cgen_state_->emitCall(
522  icmp_name(optype) + "_" + numeric_type_name(lhs_ti) +
523  null_check_suffix,
524  {lhs_lvs.front(),
525  rhs_lvs.front(),
526  cgen_state_->llInt(inline_int_null_val(lhs_ti)),
527  cgen_state_->inlineIntNull(SQLTypeInfo(kBOOLEAN, false))});
528  }
529  if (lhs_ti.get_type() == kFLOAT || lhs_ti.get_type() == kDOUBLE) {
530  return null_check_suffix.empty()
531  ? cgen_state_->ir_builder_.CreateFCmp(
532  llvm_fcmp_pred(optype), lhs_lvs.front(), rhs_lvs.front())
533  : cgen_state_->emitCall(
534  icmp_name(optype) + "_" + numeric_type_name(lhs_ti) +
535  null_check_suffix,
536  {lhs_lvs.front(),
537  rhs_lvs.front(),
538  lhs_ti.get_type() == kFLOAT ? cgen_state_->llFp(NULL_FLOAT)
539  : cgen_state_->llFp(NULL_DOUBLE),
540  cgen_state_->inlineIntNull(SQLTypeInfo(kBOOLEAN, false))});
541  }
542  CHECK(false);
543  return nullptr;
544 }
545 
546 llvm::Value* CodeGenerator::codegenQualifierCmp(const SQLOps optype,
547  const SQLQualifier qualifier,
548  std::vector<llvm::Value*> lhs_lvs,
549  const Analyzer::Expr* rhs,
550  const CompilationOptions& co) {
552  const auto& rhs_ti = rhs->get_type_info();
553  const Analyzer::Expr* arr_expr{rhs};
554  if (dynamic_cast<const Analyzer::UOper*>(rhs)) {
555  const auto cast_arr = static_cast<const Analyzer::UOper*>(rhs);
556  CHECK_EQ(kCAST, cast_arr->get_optype());
557  arr_expr = cast_arr->get_operand();
558  }
559  const auto& arr_ti = arr_expr->get_type_info();
560  const auto& elem_ti = arr_ti.get_elem_type();
561  auto rhs_lvs = codegen(arr_expr, true, co);
562  CHECK_NE(kONE, qualifier);
563  std::string fname{std::string("array_") + (qualifier == kANY ? "any" : "all") + "_" +
564  icmp_arr_name(optype)};
565  const auto& target_ti = rhs_ti.get_elem_type();
566  const bool is_real_string{target_ti.is_string() &&
567  target_ti.get_compression() != kENCODING_DICT};
568  if (is_real_string) {
569  if (g_cluster) {
570  throw std::runtime_error(
571  "Comparison between a dictionary-encoded and a none-encoded string not "
572  "supported for distributed queries");
573  }
574  if (g_enable_watchdog) {
575  throw WatchdogException(
576  "Comparison between a dictionary-encoded and a none-encoded string would be "
577  "slow");
578  }
580  throw QueryMustRunOnCpu();
581  }
582  CHECK_EQ(kENCODING_NONE, target_ti.get_compression());
583  fname += "_str";
584  }
585  if (elem_ti.is_integer() || elem_ti.is_boolean() || elem_ti.is_string() ||
586  elem_ti.is_decimal()) {
587  fname += ("_" + numeric_type_name(elem_ti));
588  } else {
589  CHECK(elem_ti.is_fp());
590  fname += elem_ti.get_type() == kDOUBLE ? "_double" : "_float";
591  }
592  if (is_real_string) {
593  CHECK_EQ(size_t(3), lhs_lvs.size());
595  fname,
597  {rhs_lvs.front(),
598  posArg(arr_expr),
599  lhs_lvs[1],
600  lhs_lvs[2],
601  cgen_state_->llInt(int64_t(executor()->getStringDictionaryProxy(
602  elem_ti.get_comp_param(), executor()->getRowSetMemoryOwner(), true))),
603  cgen_state_->inlineIntNull(elem_ti)});
604  }
605  if (target_ti.is_integer() || target_ti.is_boolean() || target_ti.is_string() ||
606  target_ti.is_decimal()) {
607  fname += ("_" + numeric_type_name(target_ti));
608  } else {
609  CHECK(target_ti.is_fp());
610  fname += target_ti.get_type() == kDOUBLE ? "_double" : "_float";
611  }
612  return cgen_state_->emitExternalCall(
613  fname,
614  get_int_type(1, cgen_state_->context_),
615  {rhs_lvs.front(),
616  posArg(arr_expr),
617  lhs_lvs.front(),
618  elem_ti.is_fp() ? static_cast<llvm::Value*>(cgen_state_->inlineFpNull(elem_ti))
619  : static_cast<llvm::Value*>(cgen_state_->inlineIntNull(elem_ti))});
620 }
void check_array_comp_cond(const Analyzer::BinOper *bin_oper)
Definition: CompareIR.cpp:182
#define CHECK_EQ(x, y)
Definition: Logger.h:211
#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
bool g_enable_watchdog
std::string icmp_name(const SQLOps op_type)
Definition: CompareIR.cpp:45
#define IS_EQUIVALENCE(X)
Definition: sqldefs.h:67
tuple d
Definition: test_fsi.py:9
CgenState * cgen_state_
#define NULL_FLOAT
SQLQualifier
Definition: sqldefs.h:69
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:319
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:228
const Expr * get_right_operand() const
Definition: Analyzer.h:443
SQLOps
Definition: sqldefs.h:29
Definition: sqldefs.h:35
Definition: sqldefs.h:36
llvm::Value * posArg(const Analyzer::Expr *) const
Definition: ColumnIR.cpp:512
Definition: sqldefs.h:38
bool get_contains_agg() const
Definition: Analyzer.h:80
Definition: sqldefs.h:49
Definition: sqldefs.h:30
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:314
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:215
int32_t intval
Definition: sqltypes.h:208
bool is_time() const
Definition: sqltypes.h:495
std::string icmp_arr_name(const SQLOps op_type)
Definition: CompareIR.cpp:64
bool g_enable_overlaps_hashjoin
Definition: Execute.cpp:96
SQLOps get_optype() const
Definition: Analyzer.h:439
llvm::LLVMContext & context_
Definition: CgenState.h:333
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:491
#define CHECK_NE(x, y)
Definition: Logger.h:212
llvm::ConstantInt * inlineIntNull(const SQLTypeInfo &)
Definition: CgenState.cpp:28
int64_t bigintval
Definition: sqltypes.h:209
bool is_timeinterval() const
Definition: sqltypes.h:500
llvm::Value * codegenFunctionOper(const Analyzer::FunctionOper *, const CompilationOptions &)
Definition: sqldefs.h:37
Definition: sqldefs.h:69
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:496
#define AUTOMATIC_IR_METADATA(CGENSTATE)
std::string get_null_check_suffix(const SQLTypeInfo &lhs_ti, const SQLTypeInfo &rhs_ti)
Definition: Execute.h:1159
const SQLTypeInfo & get_type_info() const
Definition: Analyzer.h:78
llvm::Value * emitCall(const std::string &fname, const std::vector< llvm::Value * > &args)
Definition: CgenState.cpp:174
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)
Definition: ParserNode.cpp:284
ExecutorDeviceType device_type
Definition: sqldefs.h:34
std::vector< llvm::Value * > codegen(const Analyzer::Expr *, const bool fetch_columns, const CompilationOptions &)
Definition: IRCodegen.cpp:28
Definition: sqltypes.h:51
Definition: sqldefs.h:69
HOST DEVICE EncodingType get_compression() const
Definition: sqltypes.h:322
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:546
Definition: sqldefs.h:32
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:203
bool is_geometry() const
Definition: sqltypes.h:501
Definition: sqldefs.h:31
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:33
const Expr * get_left_operand() const
Definition: Analyzer.h:442
llvm::Value * codegenCast(const Analyzer::UOper *, const CompilationOptions &)
Definition: CastIR.cpp:20
std::string numeric_type_name(const SQLTypeInfo &ti)
Definition: Execute.h:210
Definition: sqltypes.h:44
bool is_unnest(const Analyzer::Expr *expr)
Definition: Execute.h:1175
bool is_string() const
Definition: sqltypes.h:489
const std::shared_ptr< Analyzer::Expr > get_own_right_operand() const
Definition: Analyzer.h:447
bool is_decimal() const
Definition: sqltypes.h:492
const std::shared_ptr< Analyzer::Expr > get_own_left_operand() const
Definition: Analyzer.h:444
#define VLOG(n)
Definition: Logger.h:297
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:441
Executor * executor() const
#define IS_GEO_POLY(T)
Definition: sqltypes.h:249