OmniSciDB  c1a53651b2
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
CgenState Struct Reference

#include <CgenState.h>

+ Collaboration diagram for CgenState:

Classes

struct  FunctionOperValue
 
struct  HoistedLiteralLoadLocator
 

Public Types

using LiteralValue = boost::variant< int8_t, int16_t, int32_t, int64_t, float, double, std::pair< std::string, shared::StringDictKey >, std::string, std::vector< double >, std::vector< int32_t >, std::vector< int8_t >, std::pair< std::vector< int8_t >, int >>
 
using LiteralValues = std::vector< LiteralValue >
 

Public Member Functions

 CgenState (const size_t num_query_infos, const bool contains_left_deep_outer_join, Executor *executor)
 
 CgenState (const size_t num_query_infos, const bool contains_left_deep_outer_join)
 
 CgenState (llvm::LLVMContext &context)
 
std::tuple< size_t, size_t > getOrAddLiteral (const Analyzer::Constant *constant, const EncodingType enc_type, const shared::StringDictKey &dict_id, const int device_id)
 
const std::unordered_map< int,
LiteralValues > & 
getLiterals () const
 
llvm::Value * addStringConstant (const std::string &str)
 
const
StringDictionaryTranslationMgr
moveStringDictionaryTranslationMgr (std::unique_ptr< const StringDictionaryTranslationMgr > &&str_dict_translation_mgr)
 
const InValuesBitmapaddInValuesBitmap (std::unique_ptr< InValuesBitmap > &in_values_bitmap)
 
void moveInValuesBitmap (std::unique_ptr< const InValuesBitmap > &in_values_bitmap)
 
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)
 
llvm::Value * emitCall (const std::string &fname, const std::vector< llvm::Value * > &args)
 
llvm::Value * emitEntryCall (const std::string &fname, const std::vector< llvm::Value * > &args)
 
size_t getLiteralBufferUsage (const int device_id)
 
llvm::Value * castToTypeIn (llvm::Value *val, const size_t bit_width)
 
std::pair< llvm::ConstantInt
*, llvm::ConstantInt * > 
inlineIntMaxMin (const size_t byte_width, const bool is_signed)
 
llvm::ConstantInt * inlineIntNull (const SQLTypeInfo &)
 
llvm::ConstantFP * inlineFpNull (const SQLTypeInfo &)
 
llvm::Constant * inlineNull (const SQLTypeInfo &)
 
template<class T >
llvm::ConstantInt * llInt (const T v) const
 
llvm::ConstantFP * llFp (const float v) const
 
llvm::ConstantFP * llFp (const double v) const
 
llvm::ConstantInt * llBool (const bool v) const
 
void emitErrorCheck (llvm::Value *condition, llvm::Value *errorCode, std::string label)
 
std::vector< std::string > gpuFunctionsToReplace (llvm::Function *fn)
 
void replaceFunctionForGpu (const std::string &fcn_to_replace, llvm::Function *fn)
 
std::shared_ptr< ExecutorgetExecutor () const
 
llvm::LLVMContext & getExecutorContext () const
 
void set_module_shallow_copy (const std::unique_ptr< llvm::Module > &module, bool always_clone=false)
 
void maybeCloneFunctionRecursive (llvm::Function *fn)
 

Static Public Member Functions

static size_t literalBytes (const CgenState::LiteralValue &lit)
 
static size_t addAligned (const size_t off_in, const size_t alignment)
 

Public Attributes

size_t executor_id_
 
llvm::Module * module_
 
llvm::Function * row_func_
 
llvm::Function * filter_func_
 
llvm::Function * current_func_
 
llvm::BasicBlock * row_func_bb_
 
llvm::BasicBlock * filter_func_bb_
 
llvm::CallInst * row_func_call_
 
llvm::CallInst * filter_func_call_
 
std::vector< llvm::Function * > helper_functions_
 
llvm::LLVMContext & context_
 
llvm::ValueToValueMapTy vmap_
 
llvm::IRBuilder ir_builder_
 
std::unordered_map< size_t,
std::vector< llvm::Value * > > 
fetch_cache_
 
std::vector< FunctionOperValueext_call_cache_
 
std::vector< llvm::Value * > group_by_expr_cache_
 
std::vector< llvm::Value * > str_constants_
 
std::vector< llvm::Value * > frag_offsets_
 
const bool contains_left_deep_outer_join_
 
std::vector< llvm::Value * > outer_join_match_found_per_level_
 
std::unordered_map< int,
llvm::Value * > 
scan_idx_to_hash_pos_
 
InsertionOrderedMap filter_func_args_
 
std::vector< std::unique_ptr
< const InValuesBitmap > > 
in_values_bitmaps_
 
std::vector< std::unique_ptr
< const
StringDictionaryTranslationMgr > > 
str_dict_translation_mgrs_
 
std::map< std::pair
< llvm::Value *, llvm::Value * >
, ArrayLoadCodegen
array_load_cache_
 
std::unordered_map
< std::string, llvm::Value * > 
geo_target_cache_
 
bool needs_error_check_
 
bool needs_geos_
 
llvm::Function * query_func_
 
llvm::IRBuilder query_func_entry_ir_builder_
 
std::unordered_map< int,
std::vector< llvm::Value * > > 
query_func_literal_loads_
 
std::unordered_map
< llvm::Value
*, HoistedLiteralLoadLocator
row_func_hoisted_literals_
 

Private Member Functions

template<class T >
std::tuple< size_t, size_t > getOrAddLiteral (const T &val, const int device_id)
 

Private Attributes

std::unordered_map< int,
LiteralValues
literals_
 
std::unordered_map< int, size_t > literal_bytes_
 

Detailed Description

Definition at line 40 of file CgenState.h.

Member Typedef Documentation

using CgenState::LiteralValue = boost::variant<int8_t, int16_t, int32_t, int64_t, float, double, std::pair<std::string, shared::StringDictKey>, std::string, std::vector<double>, std::vector<int32_t>, std::vector<int8_t>, std::pair<std::vector<int8_t>, int>>

Definition at line 184 of file CgenState.h.

using CgenState::LiteralValues = std::vector<LiteralValue>

Definition at line 185 of file CgenState.h.

Constructor & Destructor Documentation

CgenState::CgenState ( const size_t  num_query_infos,
const bool  contains_left_deep_outer_join,
Executor executor 
)

Definition at line 25 of file CgenState.cpp.

28  : executor_id_(executor->getExecutorId())
29  , module_(nullptr)
30  , row_func_(nullptr)
31  , filter_func_(nullptr)
32  , current_func_(nullptr)
33  , row_func_bb_(nullptr)
34  , filter_func_bb_(nullptr)
35  , row_func_call_(nullptr)
36  , filter_func_call_(nullptr)
37  , context_(executor->getContext())
39  , contains_left_deep_outer_join_(contains_left_deep_outer_join)
40  , outer_join_match_found_per_level_(std::max(num_query_infos, size_t(1)) - 1)
41  , needs_error_check_(false)
42  , needs_geos_(false)
43  , query_func_(nullptr)
std::vector< llvm::Value * > outer_join_match_found_per_level_
Definition: CgenState.h:388
llvm::Function * query_func_
Definition: CgenState.h:400
llvm::IRBuilder ir_builder_
Definition: CgenState.h:377
bool needs_geos_
Definition: CgenState.h:398
const bool contains_left_deep_outer_join_
Definition: CgenState.h:387
llvm::Function * row_func_
Definition: CgenState.h:367
llvm::Module * module_
Definition: CgenState.h:366
size_t executor_id_
Definition: CgenState.h:269
llvm::LLVMContext & context_
Definition: CgenState.h:375
llvm::Function * current_func_
Definition: CgenState.h:369
llvm::CallInst * filter_func_call_
Definition: CgenState.h:373
bool needs_error_check_
Definition: CgenState.h:397
llvm::IRBuilder query_func_entry_ir_builder_
Definition: CgenState.h:401
llvm::BasicBlock * filter_func_bb_
Definition: CgenState.h:371
llvm::Function * filter_func_
Definition: CgenState.h:368
llvm::CallInst * row_func_call_
Definition: CgenState.h:372
llvm::BasicBlock * row_func_bb_
Definition: CgenState.h:370
CgenState::CgenState ( const size_t  num_query_infos,
const bool  contains_left_deep_outer_join 
)

Definition at line 46 of file CgenState.cpp.

48  : CgenState(num_query_infos,
49  contains_left_deep_outer_join,
static std::shared_ptr< Executor > getExecutor(const ExecutorId id, const std::string &debug_dir="", const std::string &debug_file="", const SystemParameters &system_parameters=SystemParameters())
Definition: Execute.cpp:475
CgenState(const size_t num_query_infos, const bool contains_left_deep_outer_join, Executor *executor)
Definition: CgenState.cpp:25
static const ExecutorId UNITARY_EXECUTOR_ID
Definition: Execute.h:373
CgenState::CgenState ( llvm::LLVMContext &  context)

Definition at line 52 of file CgenState.cpp.

54  , module_(nullptr)
55  , row_func_(nullptr)
56  , context_(context)
59  , needs_error_check_(false)
60  , needs_geos_(false)
61  , query_func_(nullptr)
llvm::Function * query_func_
Definition: CgenState.h:400
llvm::IRBuilder ir_builder_
Definition: CgenState.h:377
bool needs_geos_
Definition: CgenState.h:398
const bool contains_left_deep_outer_join_
Definition: CgenState.h:387
llvm::Function * row_func_
Definition: CgenState.h:367
llvm::Module * module_
Definition: CgenState.h:366
size_t executor_id_
Definition: CgenState.h:269
llvm::LLVMContext & context_
Definition: CgenState.h:375
bool needs_error_check_
Definition: CgenState.h:397
llvm::IRBuilder query_func_entry_ir_builder_
Definition: CgenState.h:401
static const ExecutorId INVALID_EXECUTOR_ID
Definition: Execute.h:374

Member Function Documentation

static size_t CgenState::addAligned ( const size_t  off_in,
const size_t  alignment 
)
inlinestatic

Definition at line 441 of file CgenState.h.

Referenced by getOrAddLiteral(), and Executor::serializeLiterals().

441  {
442  size_t off = off_in;
443  if (off % alignment != 0) {
444  off += (alignment - off % alignment);
445  }
446  return off + alignment;
447  }

+ Here is the caller graph for this function:

const InValuesBitmap* CgenState::addInValuesBitmap ( std::unique_ptr< InValuesBitmap > &  in_values_bitmap)
inline

Definition at line 204 of file CgenState.h.

References in_values_bitmaps_.

Referenced by CodeGenerator::codegen().

205  {
206  if (in_values_bitmap->isEmpty()) {
207  return in_values_bitmap.get();
208  }
209  in_values_bitmaps_.emplace_back(std::move(in_values_bitmap));
210  return in_values_bitmaps_.back().get();
211  }
std::vector< std::unique_ptr< const InValuesBitmap > > in_values_bitmaps_
Definition: CgenState.h:391

+ Here is the caller graph for this function:

llvm::Value* CgenState::addStringConstant ( const std::string &  str)
inline

Definition at line 189 of file CgenState.h.

References context_, get_int_type(), ir_builder_, str_constants_, and to_string().

Referenced by CodeGenerator::codegen().

189  {
190  llvm::Value* str_lv = ir_builder_.CreateGlobalString(
191  str, "str_const_" + std::to_string(std::hash<std::string>()(str)));
192  auto i8_ptr = llvm::PointerType::get(get_int_type(8, context_), 0);
193  str_constants_.push_back(str_lv);
194  str_lv = ir_builder_.CreateBitCast(str_lv, i8_ptr);
195  return str_lv;
196  }
llvm::IRBuilder ir_builder_
Definition: CgenState.h:377
std::vector< llvm::Value * > str_constants_
Definition: CgenState.h:385
llvm::Type * get_int_type(const int width, llvm::LLVMContext &context)
std::string to_string(char const *&&v)
llvm::LLVMContext & context_
Definition: CgenState.h:375

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

llvm::Value * CgenState::castToTypeIn ( llvm::Value *  val,
const size_t  bit_width 
)

Definition at line 149 of file CgenState.cpp.

References CHECK, context_, get_int_type(), and ir_builder_.

Referenced by CodeGenerator::codegenCast(), CodeGenerator::codegenCmp(), CodeGenerator::codegenFixedLengthColVarInWindow(), CodeGenerator::codegenIsNull(), CodeGenerator::codegenLogical(), CodeGenerator::codegenLogicalShortCircuit(), and anonymous_namespace{WindowFunctionIR.cpp}::get_null_value_by_size_with_encoding().

149  {
150  auto src_bits = val->getType()->getScalarSizeInBits();
151  if (src_bits == dst_bits) {
152  return val;
153  }
154  if (val->getType()->isIntegerTy()) {
155  return ir_builder_.CreateIntCast(
156  val, get_int_type(dst_bits, context_), src_bits != 1);
157  }
158  // real (not dictionary-encoded) strings; store the pointer to the payload
159  if (val->getType()->isPointerTy()) {
160  return ir_builder_.CreatePointerCast(val, get_int_type(dst_bits, context_));
161  }
162 
163  CHECK(val->getType()->isFloatTy() || val->getType()->isDoubleTy());
164 
165  llvm::Type* dst_type = nullptr;
166  switch (dst_bits) {
167  case 64:
168  dst_type = llvm::Type::getDoubleTy(context_);
169  break;
170  case 32:
171  dst_type = llvm::Type::getFloatTy(context_);
172  break;
173  default:
174  CHECK(false);
175  }
176 
177  return ir_builder_.CreateFPCast(val, dst_type);
178 }
llvm::IRBuilder ir_builder_
Definition: CgenState.h:377
llvm::Type * get_int_type(const int width, llvm::LLVMContext &context)
llvm::LLVMContext & context_
Definition: CgenState.h:375
#define CHECK(condition)
Definition: Logger.h:291

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

llvm::Value * CgenState::emitCall ( const std::string &  fname,
const std::vector< llvm::Value * > &  args 
)

Definition at line 216 of file CgenState.cpp.

References CHECK, ir_builder_, maybeCloneFunctionRecursive(), and module_.

Referenced by spatial_type::Transform::codegen(), CodeGenerator::codegen(), CodeGenerator::codegenAdd(), CodeGenerator::codegenCase(), CodeGenerator::codegenCast(), CodeGenerator::codegenCastBetweenIntTypes(), CodeGenerator::codegenCastBetweenIntTypesOverflowChecks(), CodeGenerator::codegenCastBetweenTimestamps(), CodeGenerator::codegenCastFromFp(), CodeGenerator::codegenCastToFp(), CodeGenerator::codegenCmp(), CodeGenerator::codegenConstantWidthBucketExpr(), CodeGenerator::codegenDateTruncHighPrecisionTimestamps(), CodeGenerator::codegenDiv(), CodeGenerator::codegenExtractHighPrecisionTimestamps(), CodeGenerator::codegenFpArith(), CodeGenerator::codegenFunctionOperWithCustomTypeHandling(), CodeGenerator::codegenLogical(), CodeGenerator::codegenMod(), CodeGenerator::codegenMul(), CodeGenerator::codegenSub(), CodeGenerator::codegenUMinus(), CodeGenerator::codegenWidthBucketExpr(), CodeGenerator::codegenWindowPosition(), and CodeGenerator::codgenAdjustFixedEncNull().

217  {
218  // Get the function reference from the query module.
219  auto func = module_->getFunction(fname);
220  CHECK(func) << fname;
221  // If the function called isn't external, clone the implementation from the runtime
222  // module.
224 
225  return ir_builder_.CreateCall(func, args);
226 }
void maybeCloneFunctionRecursive(llvm::Function *fn)
Definition: CgenState.cpp:180
llvm::IRBuilder ir_builder_
Definition: CgenState.h:377
llvm::Module * module_
Definition: CgenState.h:366
#define CHECK(condition)
Definition: Logger.h:291

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

llvm::Value * CgenState::emitEntryCall ( const std::string &  fname,
const std::vector< llvm::Value * > &  args 
)

Definition at line 228 of file CgenState.cpp.

References CHECK, maybeCloneFunctionRecursive(), module_, and query_func_entry_ir_builder_.

Referenced by CodeGenerator::codegenHoistedConstantsLoads().

229  {
230  // Get the function reference from the query module.
231  auto func = module_->getFunction(fname);
232  CHECK(func);
233  // If the function called isn't external, clone the implementation from the runtime
234  // module.
236 
237  return query_func_entry_ir_builder_.CreateCall(func, args);
238 }
void maybeCloneFunctionRecursive(llvm::Function *fn)
Definition: CgenState.cpp:180
llvm::Module * module_
Definition: CgenState.h:366
llvm::IRBuilder query_func_entry_ir_builder_
Definition: CgenState.h:401
#define CHECK(condition)
Definition: Logger.h:291

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

void CgenState::emitErrorCheck ( llvm::Value *  condition,
llvm::Value *  errorCode,
std::string  label 
)

Definition at line 240 of file CgenState.cpp.

References context_, current_func_, ir_builder_, and needs_error_check_.

242  {
243  needs_error_check_ = true;
244  auto check_ok = llvm::BasicBlock::Create(context_, label + "_ok", current_func_);
245  auto check_fail = llvm::BasicBlock::Create(context_, label + "_fail", current_func_);
246  ir_builder_.CreateCondBr(condition, check_ok, check_fail);
247  ir_builder_.SetInsertPoint(check_fail);
248  ir_builder_.CreateRet(errorCode);
249  ir_builder_.SetInsertPoint(check_ok);
250 }
llvm::IRBuilder ir_builder_
Definition: CgenState.h:377
llvm::LLVMContext & context_
Definition: CgenState.h:375
llvm::Function * current_func_
Definition: CgenState.h:369
bool needs_error_check_
Definition: CgenState.h:397
llvm::Value * CgenState::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 at line 396 of file CgenState.cpp.

References CHECK, CHECK_EQ, context_, ir_builder_, module_, and run_benchmark_import::result.

Referenced by spatial_type::Transform::codegen(), CodeGenerator::codegen(), CodeGenerator::codegenArrayAt(), CodeGenerator::codegenArrayBuff(), CodeGenerator::codegenArrayExpr(), CodeGenerator::codegenCastFromString(), CodeGenerator::codegenCastNonStringToString(), CodeGenerator::codegenCastTimestampToDate(), CodeGenerator::codegenCastTimestampToTime(), CodeGenerator::codegenDateTruncHighPrecisionTimestamps(), CodeGenerator::codegenFunctionOper(), CodeGenerator::codegenFunctionOperCastArgs(), CodeGenerator::codegenFunctionOperNullArg(), CodeGenerator::codegenFunctionOperWithCustomTypeHandling(), CodeGenerator::codegenGeoArgs(), CodeGenerator::codegenGeoArrayLoadAndNullcheck(), CodeGenerator::codegenGeosConstructorCall(), CodeGenerator::codegenGeosPredicateCall(), CodeGenerator::codegenIsNull(), CodeGenerator::codegenPerRowStringOper(), CodeGenerator::codegenQualifierCmp(), CodeGenerator::codegenStringFetchAndEncode(), and CodeGenerator::codegenVariableLengthStringColVar().

401  {
402  std::vector<llvm::Type*> arg_types;
403  for (const auto arg : args) {
404  CHECK(arg);
405  arg_types.push_back(arg->getType());
406  }
407  auto func_ty = llvm::FunctionType::get(ret_type, arg_types, false);
408  llvm::AttributeList attrs;
409  if (!fnattrs.empty()) {
410  std::vector<std::pair<unsigned, llvm::Attribute>> indexedAttrs;
411  indexedAttrs.reserve(fnattrs.size());
412  for (auto attr : fnattrs) {
413  indexedAttrs.emplace_back(llvm::AttributeList::FunctionIndex,
414  llvm::Attribute::get(context_, attr));
415  }
416  attrs = llvm::AttributeList::get(context_,
417  {&indexedAttrs.front(), indexedAttrs.size()});
418  }
419 
420  auto func_p = module_->getOrInsertFunction(fname, func_ty, attrs);
421  CHECK(func_p);
422  auto callee = func_p.getCallee();
423  llvm::Function* func{nullptr};
424  if (auto callee_cast = llvm::dyn_cast<llvm::ConstantExpr>(callee)) {
425  // Get or insert function automatically adds a ConstantExpr cast if the return type
426  // of the existing function does not match the supplied return type.
427  CHECK(callee_cast->isCast());
428  CHECK_EQ(callee_cast->getNumOperands(), size_t(1));
429  func = llvm::dyn_cast<llvm::Function>(callee_cast->getOperand(0));
430  } else {
431  func = llvm::dyn_cast<llvm::Function>(callee);
432  }
433  CHECK(func);
434  llvm::FunctionType* func_type = func_p.getFunctionType();
435  CHECK(func_type);
436  if (has_struct_return) {
437  const auto arg_ti = func_type->getParamType(0);
438  CHECK(arg_ti->isPointerTy() && arg_ti->getPointerElementType()->isStructTy());
439  auto attr_list = func->getAttributes();
440 #if 14 <= LLVM_VERSION_MAJOR
441  llvm::AttrBuilder arr_arg_builder(context_, attr_list.getParamAttrs(0));
442 #else
443  llvm::AttrBuilder arr_arg_builder(attr_list.getParamAttributes(0));
444 #endif
445  arr_arg_builder.addAttribute(llvm::Attribute::StructRet);
446  func->addParamAttrs(0, arr_arg_builder);
447  }
448  llvm::Value* result = ir_builder_.CreateCall(func_p, args);
449  // check the assumed type
450  CHECK_EQ(result->getType(), ret_type);
451  return result;
452 }
#define CHECK_EQ(x, y)
Definition: Logger.h:301
llvm::IRBuilder ir_builder_
Definition: CgenState.h:377
llvm::Module * module_
Definition: CgenState.h:366
llvm::LLVMContext & context_
Definition: CgenState.h:375
#define CHECK(condition)
Definition: Logger.h:291

+ Here is the caller graph for this function:

std::shared_ptr< Executor > CgenState::getExecutor ( ) const

Definition at line 372 of file CgenState.cpp.

References CHECK, executor_id_, Executor::getExecutor(), and Executor::INVALID_EXECUTOR_ID.

Referenced by getExecutorContext(), and maybeCloneFunctionRecursive().

372  {
375 }
static std::shared_ptr< Executor > getExecutor(const ExecutorId id, const std::string &debug_dir="", const std::string &debug_file="", const SystemParameters &system_parameters=SystemParameters())
Definition: Execute.cpp:475
size_t executor_id_
Definition: CgenState.h:269
static const ExecutorId INVALID_EXECUTOR_ID
Definition: Execute.h:374
#define CHECK(condition)
Definition: Logger.h:291

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

llvm::LLVMContext & CgenState::getExecutorContext ( ) const

Definition at line 377 of file CgenState.cpp.

References getExecutor().

377  {
378  return getExecutor()->getContext();
379 }
std::shared_ptr< Executor > getExecutor() const
Definition: CgenState.cpp:372

+ Here is the call graph for this function:

size_t CgenState::getLiteralBufferUsage ( const int  device_id)
inline

Definition at line 229 of file CgenState.h.

References literal_bytes_.

229 { return literal_bytes_[device_id]; }
std::unordered_map< int, size_t > literal_bytes_
Definition: CgenState.h:471
const std::unordered_map<int, LiteralValues>& CgenState::getLiterals ( ) const
inline

Definition at line 187 of file CgenState.h.

References literals_.

187 { return literals_; }
std::unordered_map< int, LiteralValues > literals_
Definition: CgenState.h:470
std::tuple<size_t, size_t> CgenState::getOrAddLiteral ( const Analyzer::Constant constant,
const EncodingType  enc_type,
const shared::StringDictKey dict_id,
const int  device_id 
)
inline

Definition at line 48 of file CgenState.h.

References Datum::bigintval, Datum::boolval, CHECK, CHECK_EQ, decimal_to_int_type(), Datum::doubleval, Datum::floatval, Analyzer::Constant::get_constval(), Analyzer::Constant::get_is_null(), Analyzer::Expr::get_type_info(), Analyzer::Constant::get_value_list(), inline_fp_null_val(), inline_int_null_val(), Datum::intval, kARRAY, kBIGINT, kBOOLEAN, kCHAR, kDATE, kDOUBLE, kENCODING_DICT, kENCODING_GEOINT, kENCODING_NONE, kFLOAT, kINT, kINTERVAL_DAY_TIME, kINTERVAL_YEAR_MONTH, kSMALLINT, kTEXT, kTIME, kTIMESTAMP, kTINYINT, kVARCHAR, Datum::smallintval, Datum::stringval, Datum::tinyintval, and run_benchmark_import::type.

Referenced by CodeGenerator::codegenHoistedConstants().

51  {
52  const auto& ti = constant->get_type_info();
53  const auto type = ti.is_decimal() ? decimal_to_int_type(ti) : ti.get_type();
54  switch (type) {
55  case kBOOLEAN:
56  return getOrAddLiteral(constant->get_is_null()
57  ? int8_t(inline_int_null_val(ti))
58  : int8_t(constant->get_constval().boolval ? 1 : 0),
59  device_id);
60  case kTINYINT:
61  return getOrAddLiteral(constant->get_is_null()
62  ? int8_t(inline_int_null_val(ti))
63  : constant->get_constval().tinyintval,
64  device_id);
65  case kSMALLINT:
66  return getOrAddLiteral(constant->get_is_null()
67  ? int16_t(inline_int_null_val(ti))
68  : constant->get_constval().smallintval,
69  device_id);
70  case kINT:
71  return getOrAddLiteral(constant->get_is_null() ? int32_t(inline_int_null_val(ti))
72  : constant->get_constval().intval,
73  device_id);
74  case kBIGINT:
75  return getOrAddLiteral(constant->get_is_null()
76  ? int64_t(inline_int_null_val(ti))
77  : constant->get_constval().bigintval,
78  device_id);
79  case kFLOAT:
80  return getOrAddLiteral(constant->get_is_null()
81  ? float(inline_fp_null_val(ti))
82  : constant->get_constval().floatval,
83  device_id);
84  case kDOUBLE:
85  return getOrAddLiteral(constant->get_is_null()
86  ? inline_fp_null_val(ti)
87  : constant->get_constval().doubleval,
88  device_id);
89  case kCHAR:
90  case kTEXT:
91  case kVARCHAR:
92  if (enc_type == kENCODING_DICT) {
93  if (constant->get_is_null()) {
94  return getOrAddLiteral(int32_t(inline_int_null_val(ti)), device_id);
95  }
96  return getOrAddLiteral(
97  std::make_pair(*constant->get_constval().stringval, dict_id), device_id);
98  }
99  CHECK_EQ(kENCODING_NONE, enc_type);
100  if (constant->get_is_null()) {
101  throw std::runtime_error(
102  "CHAR / VARCHAR NULL literal not supported in this context"); // TODO(alex):
103  // support
104  // null
105  }
106  return getOrAddLiteral(*constant->get_constval().stringval, device_id);
107  case kTIME:
108  case kTIMESTAMP:
109  case kDATE:
110  case kINTERVAL_DAY_TIME:
112  // TODO(alex): support null
113  return getOrAddLiteral(constant->get_constval().bigintval, device_id);
114  case kARRAY: {
115  if (enc_type == kENCODING_NONE) {
116  if (ti.get_subtype() == kDOUBLE) {
117  std::vector<double> double_array_literal;
118  for (const auto& value : constant->get_value_list()) {
119  const auto c = dynamic_cast<const Analyzer::Constant*>(value.get());
120  CHECK(c);
121  double d = c->get_constval().doubleval;
122  double_array_literal.push_back(d);
123  }
124  return getOrAddLiteral(double_array_literal, device_id);
125  }
126  if (ti.get_subtype() == kINT) {
127  std::vector<int32_t> int32_array_literal;
128  for (const auto& value : constant->get_value_list()) {
129  const auto c = dynamic_cast<const Analyzer::Constant*>(value.get());
130  CHECK(c);
131  int32_t i = c->get_constval().intval;
132  int32_array_literal.push_back(i);
133  }
134  return getOrAddLiteral(int32_array_literal, device_id);
135  }
136  if (ti.get_subtype() == kTINYINT) {
137  std::vector<int8_t> int8_array_literal;
138  for (const auto& value : constant->get_value_list()) {
139  const auto c = dynamic_cast<const Analyzer::Constant*>(value.get());
140  CHECK(c);
141  int8_t i = c->get_constval().tinyintval;
142  int8_array_literal.push_back(i);
143  }
144  if (ti.get_comp_param() == 64) {
145  return getOrAddLiteral(std::make_pair(int8_array_literal, 64), device_id);
146  }
147  return getOrAddLiteral(int8_array_literal, device_id);
148  }
149  throw std::runtime_error("Unsupported literal array");
150  }
151  if (enc_type == kENCODING_GEOINT) {
152  if (ti.get_subtype() == kTINYINT) {
153  std::vector<int8_t> int8_array_literal;
154  for (const auto& value : constant->get_value_list()) {
155  const auto c = dynamic_cast<const Analyzer::Constant*>(value.get());
156  CHECK(c);
157  int8_t i = c->get_constval().tinyintval;
158  int8_array_literal.push_back(i);
159  }
160  if (ti.get_comp_param() == 32) {
161  return getOrAddLiteral(std::make_pair(int8_array_literal, 32), device_id);
162  }
163  return getOrAddLiteral(int8_array_literal, device_id);
164  }
165  }
166  throw std::runtime_error("Encoded literal arrays are not supported");
167  }
168  default:
169  abort();
170  }
171  }
const std::list< std::shared_ptr< Analyzer::Expr > > & get_value_list() const
Definition: Analyzer.h:350
#define CHECK_EQ(x, y)
Definition: Logger.h:301
Definition: sqltypes.h:66
bool get_is_null() const
Definition: Analyzer.h:347
double inline_fp_null_val(const SQL_TYPE_INFO &ti)
int64_t bigintval
Definition: Datum.h:72
const SQLTypeInfo & get_type_info() const
Definition: Analyzer.h:79
std::string * stringval
Definition: Datum.h:77
SQLTypes decimal_to_int_type(const SQLTypeInfo &ti)
Definition: Datum.cpp:559
Definition: sqltypes.h:69
Definition: sqltypes.h:70
Datum get_constval() const
Definition: Analyzer.h:348
Definition: sqltypes.h:58
#define CHECK(condition)
Definition: Logger.h:291
int64_t inline_int_null_val(const SQL_TYPE_INFO &ti)
Definition: sqltypes.h:62
std::tuple< size_t, size_t > getOrAddLiteral(const Analyzer::Constant *constant, const EncodingType enc_type, const shared::StringDictKey &dict_id, const int device_id)
Definition: CgenState.h:48

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

template<class T >
std::tuple<size_t, size_t> CgenState::getOrAddLiteral ( const T &  val,
const int  device_id 
)
inlineprivate

Definition at line 453 of file CgenState.h.

References addAligned(), literal_bytes_, literalBytes(), and literals_.

453  {
454  const LiteralValue var_val(val);
455  size_t literal_found_off{0};
456  auto& literals = literals_[device_id];
457  for (const auto& literal : literals) {
458  const auto lit_bytes = literalBytes(literal);
459  literal_found_off = addAligned(literal_found_off, lit_bytes);
460  if (literal == var_val) {
461  return {literal_found_off - lit_bytes, lit_bytes};
462  }
463  }
464  literals.emplace_back(val);
465  const auto lit_bytes = literalBytes(var_val);
466  literal_bytes_[device_id] = addAligned(literal_bytes_[device_id], lit_bytes);
467  return {literal_bytes_[device_id] - lit_bytes, lit_bytes};
468  }
static size_t literalBytes(const CgenState::LiteralValue &lit)
Definition: CgenState.h:410
boost::variant< int8_t, int16_t, int32_t, int64_t, float, double, std::pair< std::string, shared::StringDictKey >, std::string, std::vector< double >, std::vector< int32_t >, std::vector< int8_t >, std::pair< std::vector< int8_t >, int >> LiteralValue
Definition: CgenState.h:184
static size_t addAligned(const size_t off_in, const size_t alignment)
Definition: CgenState.h:441
std::unordered_map< int, size_t > literal_bytes_
Definition: CgenState.h:471
std::unordered_map< int, LiteralValues > literals_
Definition: CgenState.h:470

+ Here is the call graph for this function:

std::vector< std::string > CgenState::gpuFunctionsToReplace ( llvm::Function *  fn)

Definition at line 305 of file CgenState.cpp.

References CHECK, and anonymous_namespace{CgenState.cpp}::gpu_replacement_functions.

Referenced by spatial_type::Transform::codegen().

305  {
306  std::vector<std::string> ret;
307 
308  CHECK(fn);
309  CHECK(!fn->isDeclaration());
310 
311  for (auto& basic_block : *fn) {
312  auto& inst_list = basic_block.getInstList();
313  for (auto inst_itr = inst_list.begin(); inst_itr != inst_list.end(); ++inst_itr) {
314  if (auto call_inst = llvm::dyn_cast<llvm::CallInst>(inst_itr)) {
315  auto called_fcn = call_inst->getCalledFunction();
316  CHECK(called_fcn);
317 
318  if (gpu_replacement_functions.find(called_fcn->getName().str()) !=
320  ret.emplace_back(called_fcn->getName());
321  }
322  }
323  }
324  }
325  return ret;
326 }
static const std::unordered_map< std::string, std::shared_ptr< GpuFunctionDefinition > > gpu_replacement_functions
Definition: CgenState.cpp:290
#define CHECK(condition)
Definition: Logger.h:291

+ Here is the caller graph for this function:

llvm::ConstantFP * CgenState::inlineFpNull ( const SQLTypeInfo type_info)

Definition at line 103 of file CgenState.cpp.

References CHECK, SQLTypeInfo::get_type(), SQLTypeInfo::is_fp(), kDOUBLE, kFLOAT, llFp(), NULL_DOUBLE, and NULL_FLOAT.

Referenced by CodeGenerator::codegenArrayAt(), CodeGenerator::codegenCastFromFp(), CodeGenerator::codegenCastToFp(), CodeGenerator::codegenFixedLengthColVarInWindow(), CodeGenerator::codegenUMinus(), CodeGenerator::codegenWidthBucketExpr(), CodeGenerator::endArgsNullcheck(), inlineNull(), and CodeGenerator::NullCheckCodegen::NullCheckCodegen().

103  {
104  CHECK(type_info.is_fp());
105  switch (type_info.get_type()) {
106  case kFLOAT:
107  return llFp(NULL_FLOAT);
108  case kDOUBLE:
109  return llFp(NULL_DOUBLE);
110  default:
111  abort();
112  }
113 }
#define NULL_DOUBLE
#define NULL_FLOAT
bool is_fp() const
Definition: sqltypes.h:584
HOST DEVICE SQLTypes get_type() const
Definition: sqltypes.h:381
llvm::ConstantFP * llFp(const float v) const
Definition: CgenState.h:246
#define CHECK(condition)
Definition: Logger.h:291

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

std::pair< llvm::ConstantInt *, llvm::ConstantInt * > CgenState::inlineIntMaxMin ( const size_t  byte_width,
const bool  is_signed 
)

Definition at line 120 of file CgenState.cpp.

References CHECK_EQ, context_, inline_int_max_min(), inline_uint_max_min(), and ll_int().

Referenced by CodeGenerator::codegenAdd(), CodeGenerator::codegenCastBetweenIntTypesOverflowChecks(), CodeGenerator::codegenDiv(), CodeGenerator::codegenMul(), CodeGenerator::codegenSub(), and CodeGenerator::codegenUMinus().

122  {
123  int64_t max_int{0}, min_int{0};
124  if (is_signed) {
125  std::tie(max_int, min_int) = inline_int_max_min(byte_width);
126  } else {
127  uint64_t max_uint{0}, min_uint{0};
128  std::tie(max_uint, min_uint) = inline_uint_max_min(byte_width);
129  max_int = static_cast<int64_t>(max_uint);
130  CHECK_EQ(uint64_t(0), min_uint);
131  }
132  switch (byte_width) {
133  case 1:
134  return std::make_pair(::ll_int(static_cast<int8_t>(max_int), context_),
135  ::ll_int(static_cast<int8_t>(min_int), context_));
136  case 2:
137  return std::make_pair(::ll_int(static_cast<int16_t>(max_int), context_),
138  ::ll_int(static_cast<int16_t>(min_int), context_));
139  case 4:
140  return std::make_pair(::ll_int(static_cast<int32_t>(max_int), context_),
141  ::ll_int(static_cast<int32_t>(min_int), context_));
142  case 8:
143  return std::make_pair(::ll_int(max_int, context_), ::ll_int(min_int, context_));
144  default:
145  abort();
146  }
147 }
#define CHECK_EQ(x, y)
Definition: Logger.h:301
llvm::ConstantInt * ll_int(const T v, llvm::LLVMContext &context)
llvm::LLVMContext & context_
Definition: CgenState.h:375
std::pair< uint64_t, uint64_t > inline_uint_max_min(const size_t byte_width)
std::pair< int64_t, int64_t > inline_int_max_min(const size_t byte_width)

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

llvm::ConstantInt * CgenState::inlineIntNull ( const SQLTypeInfo type_info)

Definition at line 64 of file CgenState.cpp.

References CHECK, SQLTypeInfo::get_compression(), SQLTypeInfo::get_type(), inline_int_null_val(), SQLTypeInfo::is_string(), kARRAY, kBIGINT, kBOOLEAN, kDATE, kDECIMAL, kENCODING_DICT, kENCODING_NONE, kINT, kINTERVAL_DAY_TIME, kINTERVAL_YEAR_MONTH, kNUMERIC, kSMALLINT, kTIME, kTIMESTAMP, kTINYINT, llInt(), and run_benchmark_import::type.

Referenced by CodeGenerator::codegen(), CodeGenerator::codegenArrayAt(), CodeGenerator::codegenCastBetweenIntTypes(), CodeGenerator::codegenCastBetweenIntTypesOverflowChecks(), CodeGenerator::codegenCastBetweenTimestamps(), CodeGenerator::codegenCastFromFp(), CodeGenerator::codegenCastTimestampToDate(), CodeGenerator::codegenCastTimestampToTime(), CodeGenerator::codegenCastToFp(), CodeGenerator::codegenCmp(), CodeGenerator::codegenDateTruncHighPrecisionTimestamps(), CodeGenerator::codegenDiv(), CodeGenerator::codegenExtractHighPrecisionTimestamps(), CodeGenerator::codegenFixedLengthColVarInWindow(), CodeGenerator::codegenHoistedConstantsLoads(), CodeGenerator::codegenIntConst(), CodeGenerator::codegenIsNullNumber(), CodeGenerator::codegenLogical(), CodeGenerator::codegenLogicalShortCircuit(), CodeGenerator::codegenQualifierCmp(), CodeGenerator::codegenUMinus(), CodeGenerator::codgenAdjustFixedEncNull(), CodeGenerator::endArgsNullcheck(), inlineNull(), and CodeGenerator::NullCheckCodegen::NullCheckCodegen().

64  {
65  auto type = type_info.get_type();
66  if (type_info.is_string()) {
67  switch (type_info.get_compression()) {
68  case kENCODING_DICT:
69  return llInt(static_cast<int32_t>(inline_int_null_val(type_info)));
70  case kENCODING_NONE:
71  return llInt(int64_t(0));
72  default:
73  CHECK(false);
74  }
75  }
76  switch (type) {
77  case kBOOLEAN:
78  return llInt(static_cast<int8_t>(inline_int_null_val(type_info)));
79  case kTINYINT:
80  return llInt(static_cast<int8_t>(inline_int_null_val(type_info)));
81  case kSMALLINT:
82  return llInt(static_cast<int16_t>(inline_int_null_val(type_info)));
83  case kINT:
84  return llInt(static_cast<int32_t>(inline_int_null_val(type_info)));
85  case kBIGINT:
86  return llInt(static_cast<int64_t>(inline_int_null_val(type_info)));
87  case kTIME:
88  case kTIMESTAMP:
89  case kDATE:
90  case kINTERVAL_DAY_TIME:
92  return llInt(inline_int_null_val(type_info));
93  case kDECIMAL:
94  case kNUMERIC:
95  return llInt(inline_int_null_val(type_info));
96  case kARRAY:
97  return llInt(int64_t(0));
98  default:
99  abort();
100  }
101 }
Definition: sqltypes.h:66
HOST DEVICE SQLTypes get_type() const
Definition: sqltypes.h:381
Definition: sqltypes.h:70
HOST DEVICE EncodingType get_compression() const
Definition: sqltypes.h:389
llvm::ConstantInt * llInt(const T v) const
Definition: CgenState.h:242
#define CHECK(condition)
Definition: Logger.h:291
int64_t inline_int_null_val(const SQL_TYPE_INFO &ti)
Definition: sqltypes.h:62
bool is_string() const
Definition: sqltypes.h:580

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

llvm::Constant * CgenState::inlineNull ( const SQLTypeInfo ti)

Definition at line 115 of file CgenState.cpp.

References inlineFpNull(), inlineIntNull(), and SQLTypeInfo::is_fp().

Referenced by CodeGenerator::codegenPerRowStringOper().

115  {
116  return ti.is_fp() ? static_cast<llvm::Constant*>(inlineFpNull(ti))
117  : static_cast<llvm::Constant*>(inlineIntNull(ti));
118 }
bool is_fp() const
Definition: sqltypes.h:584
llvm::ConstantInt * inlineIntNull(const SQLTypeInfo &)
Definition: CgenState.cpp:64
llvm::ConstantFP * inlineFpNull(const SQLTypeInfo &)
Definition: CgenState.cpp:103

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

static size_t CgenState::literalBytes ( const CgenState::LiteralValue lit)
inlinestatic

Definition at line 410 of file CgenState.h.

Referenced by CodeGenerator::codegenHoistedConstantsLoads(), getOrAddLiteral(), and Executor::serializeLiterals().

410  {
411  switch (lit.which()) {
412  case 0:
413  return 1; // int8_t
414  case 1:
415  return 2; // int16_t
416  case 2:
417  return 4; // int32_t
418  case 3:
419  return 8; // int64_t
420  case 4:
421  return 4; // float
422  case 5:
423  return 8; // double
424  case 6:
425  return 4; // std::pair<std::string, int>
426  case 7:
427  return 4; // std::string
428  case 8:
429  return 4; // std::vector<double>
430  case 9:
431  return 4; // std::vector<int32_t>
432  case 10:
433  return 4; // std::vector<int8_t>
434  case 11:
435  return 4; // std::pair<std::vector<int8_t>, int>
436  default:
437  abort();
438  }
439  }

+ Here is the caller graph for this function:

llvm::ConstantInt* CgenState::llBool ( const bool  v) const
inline

Definition at line 256 of file CgenState.h.

References context_, and ll_bool().

Referenced by Executor::buildJoinLoops(), CodeGenerator::codegenGeosConstructorCall(), CodeGenerator::codegenGeosPredicateCall(), and CodeGenerator::NullCheckCodegen::NullCheckCodegen().

256 { return ::ll_bool(v, context_); }
llvm::LLVMContext & context_
Definition: CgenState.h:375
llvm::ConstantInt * ll_bool(const bool v, llvm::LLVMContext &context)

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

llvm::ConstantFP* CgenState::llFp ( const float  v) const
inline

Definition at line 246 of file CgenState.h.

References context_.

Referenced by CodeGenerator::codegenArrayExpr(), CodeGenerator::codegenCmp(), CodeGenerator::codegenDiv(), CodeGenerator::codegenFpArith(), CodeGenerator::codegenGeoUOper(), CodeGenerator::codegenIsNullNumber(), anonymous_namespace{WindowFunctionIR.cpp}::get_null_value_by_size(), anonymous_namespace{WindowFunctionIR.cpp}::get_null_value_by_size_with_encoding(), and inlineFpNull().

246  {
247  return static_cast<llvm::ConstantFP*>(
248  llvm::ConstantFP::get(llvm::Type::getFloatTy(context_), v));
249  }
llvm::LLVMContext & context_
Definition: CgenState.h:375

+ Here is the caller graph for this function:

llvm::ConstantFP* CgenState::llFp ( const double  v) const
inline

Definition at line 251 of file CgenState.h.

References context_.

251  {
252  return static_cast<llvm::ConstantFP*>(
253  llvm::ConstantFP::get(llvm::Type::getDoubleTy(context_), v));
254  }
llvm::LLVMContext & context_
Definition: CgenState.h:375
template<class T >
llvm::ConstantInt* CgenState::llInt ( const T  v) const
inline

Definition at line 242 of file CgenState.h.

References context_, and ll_int().

Referenced by spatial_type::Transform::codegen(), CodeGenerator::codegen(), CodeGenerator::codegenAdd(), CodeGenerator::codegenArrayExpr(), CodeGenerator::codegenBinOpWithOverflowForCPU(), CodeGenerator::codegenCastBetweenIntTypes(), CodeGenerator::codegenCastBetweenIntTypesOverflowChecks(), CodeGenerator::codegenCastBetweenTimestamps(), CodeGenerator::codegenCastFromString(), CodeGenerator::codegenCastNonStringToString(), CodeGenerator::codegenCastTimestampToDate(), CodeGenerator::codegenCastTimestampToTime(), CodeGenerator::codegenCmp(), CodeGenerator::codegenCompression(), CodeGenerator::codegenDateTruncHighPrecisionTimestamps(), CodeGenerator::codegenDiv(), CodeGenerator::codegenExtractHighPrecisionTimestamps(), CodeGenerator::codegenFixedLengthColVarInWindow(), CodeGenerator::codegenFunctionOper(), CodeGenerator::codegenFunctionOperCastArgs(), CodeGenerator::codegenFunctionOperWithCustomTypeHandling(), CodeGenerator::codegenGeoArgs(), CodeGenerator::codegenGeoBinOper(), CodeGenerator::codegenGeosConstructorCall(), CodeGenerator::codegenGeosPredicateCall(), CodeGenerator::codegenGeoUOper(), CodeGenerator::codegenHoistedConstantsLoads(), CodeGenerator::codegenHoistedConstantsPlaceholders(), CodeGenerator::codegenIntConst(), CodeGenerator::codegenMod(), CodeGenerator::codegenMul(), CodeGenerator::codegenPerRowStringOper(), CodeGenerator::codegenQualifierCmp(), CodeGenerator::codegenRowId(), CodeGenerator::codegenStringFetchAndEncode(), CodeGenerator::codegenSub(), CodeGenerator::codegenUMinus(), CodeGenerator::codegenWidthBucketExpr(), CodeGenerator::codegenWindowPosition(), CodeGenerator::codgenAdjustFixedEncNull(), anonymous_namespace{WindowFunctionIR.cpp}::get_null_value_by_size(), anonymous_namespace{WindowFunctionIR.cpp}::get_null_value_by_size_with_encoding(), and inlineIntNull().

242  {
244  }
llvm::ConstantInt * ll_int(const T v, llvm::LLVMContext &context)
llvm::LLVMContext & context_
Definition: CgenState.h:375

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

void CgenState::maybeCloneFunctionRecursive ( llvm::Function *  fn)

Definition at line 180 of file CgenState.cpp.

References CHECK, getExecutor(), and vmap_.

Referenced by spatial_type::Transform::codegen(), emitCall(), and emitEntryCall().

180  {
181  CHECK(fn);
182  if (!fn->isDeclaration()) {
183  return;
184  }
185 
186  // Get the implementation from the runtime module.
187  auto func_impl = getExecutor()->get_rt_module()->getFunction(fn->getName());
188  CHECK(func_impl) << fn->getName().str();
189 
190  if (func_impl->isDeclaration()) {
191  return;
192  }
193 
194  auto DestI = fn->arg_begin();
195  for (auto arg_it = func_impl->arg_begin(); arg_it != func_impl->arg_end(); ++arg_it) {
196  DestI->setName(arg_it->getName());
197  vmap_[&*arg_it] = &*DestI++;
198  }
199 
200  llvm::SmallVector<llvm::ReturnInst*, 8> Returns; // Ignore returns cloned.
201 #if LLVM_VERSION_MAJOR > 12
202  llvm::CloneFunctionInto(
203  fn, func_impl, vmap_, llvm::CloneFunctionChangeType::DifferentModule, Returns);
204 #else
205  llvm::CloneFunctionInto(fn, func_impl, vmap_, /*ModuleLevelChanges=*/true, Returns);
206 #endif
207 
208  for (auto it = llvm::inst_begin(fn), e = llvm::inst_end(fn); it != e; ++it) {
209  if (llvm::isa<llvm::CallInst>(*it)) {
210  auto& call = llvm::cast<llvm::CallInst>(*it);
211  maybeCloneFunctionRecursive(call.getCalledFunction());
212  }
213  }
214 }
std::shared_ptr< Executor > getExecutor() const
Definition: CgenState.cpp:372
void maybeCloneFunctionRecursive(llvm::Function *fn)
Definition: CgenState.cpp:180
#define CHECK(condition)
Definition: Logger.h:291
llvm::ValueToValueMapTy vmap_
Definition: CgenState.h:376

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

void CgenState::moveInValuesBitmap ( std::unique_ptr< const InValuesBitmap > &  in_values_bitmap)
inline

Definition at line 212 of file CgenState.h.

References in_values_bitmaps_.

212  {
213  if (!in_values_bitmap->isEmpty()) {
214  in_values_bitmaps_.emplace_back(std::move(in_values_bitmap));
215  }
216  }
std::vector< std::unique_ptr< const InValuesBitmap > > in_values_bitmaps_
Definition: CgenState.h:391
const StringDictionaryTranslationMgr* CgenState::moveStringDictionaryTranslationMgr ( std::unique_ptr< const StringDictionaryTranslationMgr > &&  str_dict_translation_mgr)
inline

Definition at line 198 of file CgenState.h.

References str_dict_translation_mgrs_.

Referenced by CodeGenerator::codegen(), CodeGenerator::codegenCastFromString(), and CodeGenerator::codegenPseudoStringOper().

199  {
200  str_dict_translation_mgrs_.emplace_back(std::move(str_dict_translation_mgr));
201  return str_dict_translation_mgrs_.back().get();
202  }
std::vector< std::unique_ptr< const StringDictionaryTranslationMgr > > str_dict_translation_mgrs_
Definition: CgenState.h:393

+ Here is the caller graph for this function:

void CgenState::replaceFunctionForGpu ( const std::string &  fcn_to_replace,
llvm::Function *  fn 
)

Definition at line 328 of file CgenState.cpp.

References run_benchmark_import::args, CHECK, CHECK_EQ, context_, anonymous_namespace{CgenState.cpp}::gpu_replacement_functions, module_, and VLOG.

Referenced by spatial_type::Transform::codegen().

329  {
330  CHECK(fn);
331  CHECK(!fn->isDeclaration());
332 
333  auto map_it = gpu_replacement_functions.find(fcn_to_replace);
334  if (map_it == gpu_replacement_functions.end()) {
335  throw QueryMustRunOnCpu("Codegen failed: Could not find replacement functon for " +
336  fcn_to_replace +
337  " to run on gpu. Query step must run in cpu mode.");
338  }
339  const auto& gpu_fcn_obj = map_it->second;
340  CHECK(gpu_fcn_obj);
341  VLOG(1) << "Replacing " << fcn_to_replace << " with " << gpu_fcn_obj->name_
342  << " for parent function " << fn->getName().str();
343 
344  for (auto& basic_block : *fn) {
345  auto& inst_list = basic_block.getInstList();
346  for (auto inst_itr = inst_list.begin(); inst_itr != inst_list.end(); ++inst_itr) {
347  if (auto call_inst = llvm::dyn_cast<llvm::CallInst>(inst_itr)) {
348  auto called_fcn = call_inst->getCalledFunction();
349  CHECK(called_fcn);
350 
351  if (called_fcn->getName() == fcn_to_replace) {
352  std::vector<llvm::Value*> args;
353  std::vector<llvm::Type*> arg_types;
354  for (auto& arg : call_inst->args()) {
355  arg_types.push_back(arg.get()->getType());
356  args.push_back(arg.get());
357  }
358  auto gpu_func = gpu_fcn_obj->getFunction(module_, context_);
359  CHECK(gpu_func);
360  auto gpu_func_type = gpu_func.getFunctionType();
361  CHECK(gpu_func_type);
362  CHECK_EQ(gpu_func_type->getReturnType(), called_fcn->getReturnType());
363  llvm::ReplaceInstWithInst(call_inst,
364  llvm::CallInst::Create(gpu_func, args, ""));
365  return;
366  }
367  }
368  }
369  }
370 }
#define CHECK_EQ(x, y)
Definition: Logger.h:301
static const std::unordered_map< std::string, std::shared_ptr< GpuFunctionDefinition > > gpu_replacement_functions
Definition: CgenState.cpp:290
llvm::Module * module_
Definition: CgenState.h:366
llvm::LLVMContext & context_
Definition: CgenState.h:375
#define CHECK(condition)
Definition: Logger.h:291
#define VLOG(n)
Definition: Logger.h:387

+ Here is the caller graph for this function:

void CgenState::set_module_shallow_copy ( const std::unique_ptr< llvm::Module > &  module,
bool  always_clone = false 
)

Definition at line 381 of file CgenState.cpp.

References CodeGenerator::alwaysCloneRuntimeFunction(), module_, and vmap_.

Referenced by GpuReductionHelperJIT::codegen().

382  {
383  module_ =
384  llvm::CloneModule(*llvm_module, vmap_, [always_clone](const llvm::GlobalValue* gv) {
385  auto func = llvm::dyn_cast<llvm::Function>(gv);
386  if (!func) {
387  return true;
388  }
389  return (func->getLinkage() == llvm::GlobalValue::LinkageTypes::PrivateLinkage ||
390  func->getLinkage() == llvm::GlobalValue::LinkageTypes::InternalLinkage ||
391  (always_clone && CodeGenerator::alwaysCloneRuntimeFunction(func)));
392  }).release();
393 }
llvm::Module * module_
Definition: CgenState.h:366
static bool alwaysCloneRuntimeFunction(const llvm::Function *func)
llvm::ValueToValueMapTy vmap_
Definition: CgenState.h:376

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

Member Data Documentation

std::map<std::pair<llvm::Value*, llvm::Value*>, ArrayLoadCodegen> CgenState::array_load_cache_

Definition at line 395 of file CgenState.h.

Referenced by CodeGenerator::codegenGeoArrayLoadAndNullcheck().

const bool CgenState::contains_left_deep_outer_join_

Definition at line 387 of file CgenState.h.

llvm::LLVMContext& CgenState::context_

Definition at line 375 of file CgenState.h.

Referenced by addStringConstant(), CodeGenerator::beginArgsNullcheck(), CodeGenerator::castArrayPointer(), castToTypeIn(), spatial_type::Transform::codegen(), CodeGenerator::codegen(), CodeGenerator::codegenAdd(), CodeGenerator::codegenArrayAt(), CodeGenerator::codegenArrayBuff(), CodeGenerator::codegenArrayExpr(), CodeGenerator::codegenBinOpWithOverflowForCPU(), CodeGenerator::codegenBufferArgs(), CodeGenerator::codegenCase(), CodeGenerator::codegenCast(), CodeGenerator::codegenCastBetweenIntTypes(), CodeGenerator::codegenCastBetweenIntTypesOverflowChecks(), CodeGenerator::codegenCastFromFp(), CodeGenerator::codegenCastFromString(), CodeGenerator::codegenCastNonStringToString(), CodeGenerator::codegenCastTimestampToDate(), CodeGenerator::codegenCastTimestampToTime(), CodeGenerator::codegenCastToFp(), CodeGenerator::codegenCmp(), CodeGenerator::codegenDateTruncHighPrecisionTimestamps(), CodeGenerator::codegenDiv(), CodeGenerator::codegenFixedLengthColVar(), CodeGenerator::codegenFixedLengthColVarInWindow(), CodeGenerator::codegenFunctionOper(), CodeGenerator::codegenFunctionOperCastArgs(), CodeGenerator::codegenFunctionOperNullArg(), CodeGenerator::codegenFunctionOperWithCustomTypeHandling(), CodeGenerator::codegenGeoArgs(), CodeGenerator::codegenGeoArrayLoadAndNullcheck(), CodeGenerator::codegenGeoBinOper(), CodeGenerator::codegenGeoConstant(), CodeGenerator::codegenGeosConstructorCall(), CodeGenerator::codegenGeosPredicateCall(), CodeGenerator::codegenGeoUOper(), CodeGenerator::codegenHoistedConstantsLoads(), CodeGenerator::codegenIsNull(), spatial_type::Transform::codegenLoads(), CodeGenerator::codegenLogicalShortCircuit(), CodeGenerator::codegenMod(), CodeGenerator::codegenMul(), CodeGenerator::codegenOuterJoinNullPlaceholder(), CodeGenerator::codegenOverlaps(), CodeGenerator::codegenPerRowStringOper(), CodeGenerator::codegenQualifierCmp(), CodeGenerator::codegenSkipOverflowCheckForNull(), CodeGenerator::codegenStringFetchAndEncode(), CodeGenerator::codegenSub(), CodeGenerator::codegenUMinus(), CodeGenerator::codegenVariableLengthStringColVar(), CodeGenerator::codegenWidthBucketExpr(), CodeGenerator::codgenAdjustFixedEncNull(), CodeGenerator::colByteStream(), anonymous_namespace{ResultSetReductionCodegen.cpp}::create_entry_block(), anonymous_namespace{ResultSetReductionJIT.cpp}::create_llvm_function(), anonymous_namespace{ResultSetReductionInterpreterStubs.cpp}::create_stub_function(), CodeGenerator::createLineStringStructType(), CodeGenerator::createMultiLineStringStructType(), CodeGenerator::createMultiPointStructType(), CodeGenerator::createMultiPolygonStructType(), CodeGenerator::createPointStructType(), CodeGenerator::createPolygonStructType(), CodeGenerator::createStringViewStructType(), emitErrorCheck(), emitExternalCall(), CodeGenerator::endArgsNullcheck(), anonymous_namespace{TableFunctionCompilationContext.cpp}::generate_entry_point(), anonymous_namespace{ExtensionsIR.cpp}::get_buffer_struct_type(), inlineIntMaxMin(), llBool(), llFp(), llInt(), AutomaticIRMetadataGuard::markInstructions(), CodeGenerator::NullCheckCodegen::NullCheckCodegen(), CodeGenerator::posArg(), and replaceFunctionForGpu().

size_t CgenState::executor_id_

Definition at line 269 of file CgenState.h.

Referenced by getExecutor().

std::vector<FunctionOperValue> CgenState::ext_call_cache_

Definition at line 383 of file CgenState.h.

Referenced by CodeGenerator::codegenFunctionOper().

std::unordered_map<size_t, std::vector<llvm::Value*> > CgenState::fetch_cache_
llvm::Function* CgenState::filter_func_

Definition at line 368 of file CgenState.h.

InsertionOrderedMap CgenState::filter_func_args_

Definition at line 390 of file CgenState.h.

llvm::BasicBlock* CgenState::filter_func_bb_

Definition at line 371 of file CgenState.h.

llvm::CallInst* CgenState::filter_func_call_

Definition at line 373 of file CgenState.h.

std::vector<llvm::Value*> CgenState::frag_offsets_

Definition at line 386 of file CgenState.h.

Referenced by CodeGenerator::codegenColVar(), and CodeGenerator::codegenRowId().

std::unordered_map<std::string, llvm::Value*> CgenState::geo_target_cache_

Definition at line 396 of file CgenState.h.

Referenced by CodeGenerator::codegenGeoOperator().

std::vector<llvm::Value*> CgenState::group_by_expr_cache_

Definition at line 384 of file CgenState.h.

Referenced by CodeGenerator::resolveGroupedColumnReference().

std::vector<llvm::Function*> CgenState::helper_functions_

Definition at line 374 of file CgenState.h.

std::vector<std::unique_ptr<const InValuesBitmap> > CgenState::in_values_bitmaps_

Definition at line 391 of file CgenState.h.

Referenced by addInValuesBitmap(), and moveInValuesBitmap().

llvm::IRBuilder CgenState::ir_builder_

Definition at line 377 of file CgenState.h.

Referenced by addStringConstant(), CodeGenerator::beginArgsNullcheck(), Executor::buildJoinLoops(), CodeGenerator::castArrayPointer(), castToTypeIn(), JoinLoop::codegen(), spatial_type::Transform::codegen(), CodeGenerator::codegen(), CodeGenerator::codegenAdd(), CodeGenerator::codegenArrayAt(), CodeGenerator::codegenArrayExpr(), CodeGenerator::codegenBinOpWithOverflowForCPU(), CodeGenerator::codegenBufferArgs(), CodeGenerator::codegenCase(), CodeGenerator::codegenCast(), CodeGenerator::codegenCastBetweenIntTypes(), CodeGenerator::codegenCastBetweenIntTypesOverflowChecks(), CodeGenerator::codegenCastBetweenTimestamps(), CodeGenerator::codegenCastFromFp(), CodeGenerator::codegenCastToFp(), CodeGenerator::codegenCmp(), CodeGenerator::codegenColVar(), CodeGenerator::codegenDateTruncHighPrecisionTimestamps(), CodeGenerator::codegenDiv(), CodeGenerator::codegenExtractHighPrecisionTimestamps(), CodeGenerator::codegenFixedLengthColVar(), CodeGenerator::codegenFixedLengthColVarInWindow(), CodeGenerator::codegenFpArith(), CodeGenerator::codegenFunctionOper(), CodeGenerator::codegenFunctionOperCastArgs(), CodeGenerator::codegenFunctionOperNullArg(), CodeGenerator::codegenFunctionOperWithCustomTypeHandling(), CodeGenerator::codegenGeoArgs(), CodeGenerator::codegenGeoConstant(), CodeGenerator::codegenGeoLineStringArgs(), CodeGenerator::codegenGeoMultiLineStringArgs(), CodeGenerator::codegenGeoMultiPointArgs(), CodeGenerator::codegenGeoMultiPolygonArgs(), CodeGenerator::codegenGeoPointArgs(), CodeGenerator::codegenGeoPolygonArgs(), CodeGenerator::codegenGeosConstructorCall(), CodeGenerator::codegenGeosPredicateCall(), CodeGenerator::codegenHoistedConstantsPlaceholders(), CodeGenerator::codegenIsNull(), CodeGenerator::codegenIsNullNumber(), spatial_type::Transform::codegenLoads(), CodeGenerator::codegenLogical(), CodeGenerator::codegenLogicalShortCircuit(), CodeGenerator::codegenMod(), CodeGenerator::codegenMul(), CodeGenerator::codegenOuterJoinNullPlaceholder(), CodeGenerator::codegenRowId(), CodeGenerator::codegenSkipOverflowCheckForNull(), CodeGenerator::codegenStringFetchAndEncode(), CodeGenerator::codegenSub(), CodeGenerator::codegenUMinus(), CodeGenerator::codegenVariableLengthStringColVar(), CodeGenerator::codegenWidthBucketExpr(), CodeGenerator::codgenAdjustFixedEncNull(), ScalarCodeGenerator::compile(), anonymous_namespace{ResultSetReductionCodegen.cpp}::create_entry_block(), anonymous_namespace{ResultSetReductionInterpreterStubs.cpp}::create_stub_function(), emitCall(), emitErrorCheck(), emitExternalCall(), CodeGenerator::endArgsNullcheck(), JoinLoop::evaluateOuterJoinCondition(), CodeGenerator::NullCheckCodegen::NullCheckCodegen(), CodeGenerator::posArg(), and CodeGenerator::toBool().

std::unordered_map<int, size_t> CgenState::literal_bytes_
private

Definition at line 471 of file CgenState.h.

Referenced by getLiteralBufferUsage(), and getOrAddLiteral().

std::unordered_map<int, LiteralValues> CgenState::literals_
private

Definition at line 470 of file CgenState.h.

Referenced by getLiterals(), and getOrAddLiteral().

bool CgenState::needs_geos_
std::vector<llvm::Value*> CgenState::outer_join_match_found_per_level_
llvm::Function* CgenState::query_func_

Definition at line 400 of file CgenState.h.

Referenced by CodeGenerator::codegenHoistedConstantsLoads().

llvm::IRBuilder CgenState::query_func_entry_ir_builder_

Definition at line 401 of file CgenState.h.

Referenced by CodeGenerator::codegenHoistedConstantsLoads(), and emitEntryCall().

std::unordered_map<int, std::vector<llvm::Value*> > CgenState::query_func_literal_loads_

Definition at line 402 of file CgenState.h.

Referenced by CodeGenerator::codegenHoistedConstants().

llvm::BasicBlock* CgenState::row_func_bb_

Definition at line 370 of file CgenState.h.

llvm::CallInst* CgenState::row_func_call_

Definition at line 372 of file CgenState.h.

std::unordered_map<llvm::Value*, HoistedLiteralLoadLocator> CgenState::row_func_hoisted_literals_

Definition at line 408 of file CgenState.h.

Referenced by CodeGenerator::codegenHoistedConstantsPlaceholders().

std::unordered_map<int, llvm::Value*> CgenState::scan_idx_to_hash_pos_

Definition at line 389 of file CgenState.h.

Referenced by CodeGenerator::posArg().

std::vector<llvm::Value*> CgenState::str_constants_

Definition at line 385 of file CgenState.h.

Referenced by addStringConstant().

std::vector<std::unique_ptr<const StringDictionaryTranslationMgr> > CgenState::str_dict_translation_mgrs_

Definition at line 393 of file CgenState.h.

Referenced by moveStringDictionaryTranslationMgr().

llvm::ValueToValueMapTy CgenState::vmap_

The documentation for this struct was generated from the following files: