GpuReductionHelperJIT Class Reference

#include <ResultSetReductionJIT.h>

Inheritance diagram for GpuReductionHelperJIT:

Collaboration diagram for GpuReductionHelperJIT:

Public Member Functions
	GpuReductionHelperJIT (const QueryMemoryDescriptor &query_mem_desc, const std::vector< TargetInfo > &targets, const std::vector< int64_t > &target_init_vals, const size_t executor_id)
ReductionCode	codegen () const override
Public Member Functions inherited from ResultSetReductionJIT
	ResultSetReductionJIT (const QueryMemoryDescriptor &query_mem_desc, const std::vector< TargetInfo > &targets, const std::vector< int64_t > &target_init_vals, const size_t executor_id)
virtual	~ResultSetReductionJIT ()=default

Private Attributes
const QueryMemoryDescriptor &	query_mem_desc_

Additional Inherited Members
Protected Member Functions inherited from ResultSetReductionJIT
void	isEmpty (const ReductionCode &reduction_code) const
void	reduceOneEntryNoCollisions (const ReductionCode &reduction_code) const
void	reduceOneEntryNoCollisionsIdx (const ReductionCode &reduction_code) const
void	reduceLoop (const ReductionCode &reduction_code) const
Protected Attributes inherited from ResultSetReductionJIT
size_t	executor_id_

Detailed Description

This is a helper class for performing GPU reduction code. It uses the same functions that ResultSetReductionJIT generates so that it can be reused and help the reduction procedure within GPU.

Definition at line 150 of file ResultSetReductionJIT.h.

Constructor & Destructor Documentation

GpuReductionHelperJIT::GpuReductionHelperJIT ( const QueryMemoryDescriptor &  query_mem_desc, const std::vector< TargetInfo > &  targets, const std::vector< int64_t > &  target_init_vals, const size_t  executor_id  )

inline

Definition at line 152 of file ResultSetReductionJIT.h.

References CHECK, QueryMemoryDescriptor::didOutputColumnar(), QueryMemoryDescriptor::getQueryDescriptionType(), heavyai::GroupByPerfectHash, QueryMemoryDescriptor::hasKeylessHash(), and query_mem_desc_.

      : ResultSetReductionJIT(query_mem_desc, targets, target_init_vals, executor_id)
      , query_mem_desc_(query_mem_desc) {
    CHECK(query_mem_desc_.getQueryDescriptionType() ==
          QueryDescriptionType::GroupByPerfectHash);
    CHECK(!query_mem_desc_.didOutputColumnar());
    CHECK(query_mem_desc_.hasKeylessHash());
  }

Here is the call graph for this function:

Member Function Documentation

ReductionCode GpuReductionHelperJIT::codegen ( ) const

overridevirtual

generates code for perfect hash group by reduction: the following functions are internally created: isEmpty, reduceOneEntryNoCollision (reduce for perfect hash), reduceOneEntryNoCollissionsIdx(reduce one slot for perfect hash), and reduceLoop (the outer loop).

Reimplemented from ResultSetReductionJIT.

Definition at line 1327 of file ResultSetReductionJIT.cpp.

References AUTOMATIC_IR_METADATA, ReductionCode::cgen_state, CHECK, anonymous_namespace{ResultSetReductionJIT.cpp}::create_llvm_function(), ResultSetReductionJIT::executor_id_, f(), Executor::getExecutor(), QueryMemoryDescriptor::getQueryDescriptionType(), heavyai::GroupByPerfectHash, ReductionCode::ir_is_empty, ReductionCode::ir_reduce_loop, ReductionCode::ir_reduce_one_entry, ReductionCode::ir_reduce_one_entry_idx, ResultSetReductionJIT::isEmpty(), ReductionCode::module, ResultSetReductionJIT::query_mem_desc_, ResultSetReductionJIT::reduceLoop(), ResultSetReductionJIT::reduceOneEntryNoCollisions(), ResultSetReductionJIT::reduceOneEntryNoCollisionsIdx(), CgenState::set_module_shallow_copy(), anonymous_namespace{ResultSetReductionJIT.cpp}::setup_functions_ir(), and translate_function().

                                                   {
  const auto hash_type = query_mem_desc_.getQueryDescriptionType();
  auto reduction_code = setup_functions_ir(hash_type);
  CHECK(hash_type == QueryDescriptionType::GroupByPerfectHash);
  isEmpty(reduction_code);
  reduceOneEntryNoCollisions(reduction_code);
  reduceOneEntryNoCollisionsIdx(reduction_code);
  reduceLoop(reduction_code);
  auto executor = Executor::getExecutor(executor_id_);
  auto cgen_state_ = std::unique_ptr<CgenState>(new CgenState({}, false, executor.get()));
  auto cgen_state = reduction_code.cgen_state = cgen_state_.get();
  // CHECK(executor->thread_id_ == logger::thread_id());  // do we need compilation mutex?
  cgen_state->set_module_shallow_copy(executor->get_rt_module());
  reduction_code.module = cgen_state->module_;

  AUTOMATIC_IR_METADATA(cgen_state);
  auto ir_is_empty = create_llvm_function(reduction_code.ir_is_empty.get(), cgen_state);
  auto ir_reduce_one_entry =
      create_llvm_function(reduction_code.ir_reduce_one_entry.get(), cgen_state);
  auto ir_reduce_one_entry_idx =
      create_llvm_function(reduction_code.ir_reduce_one_entry_idx.get(), cgen_state);
  auto ir_reduce_loop =
      create_llvm_function(reduction_code.ir_reduce_loop.get(), cgen_state);
  std::unordered_map<const Function*, llvm::Function*> f;
  f.emplace(reduction_code.ir_is_empty.get(), ir_is_empty);
  f.emplace(reduction_code.ir_reduce_one_entry.get(), ir_reduce_one_entry);
  f.emplace(reduction_code.ir_reduce_one_entry_idx.get(), ir_reduce_one_entry_idx);
  f.emplace(reduction_code.ir_reduce_loop.get(), ir_reduce_loop);
  translate_function(reduction_code.ir_is_empty.get(), ir_is_empty, reduction_code, f);
  translate_function(
      reduction_code.ir_reduce_one_entry.get(), ir_reduce_one_entry, reduction_code, f);
  translate_function(reduction_code.ir_reduce_one_entry_idx.get(),
                     ir_reduce_one_entry_idx,
                     reduction_code,
                     f);
  translate_function(
      reduction_code.ir_reduce_loop.get(), ir_reduce_loop, reduction_code, f);
  reduction_code.llvm_reduce_loop = ir_reduce_loop;
  reduction_code.cgen_state = nullptr;
  return reduction_code;
}

Here is the call graph for this function:

Member Data Documentation

const QueryMemoryDescriptor& GpuReductionHelperJIT::query_mem_desc_

private

Definition at line 172 of file ResultSetReductionJIT.h.

Referenced by GpuReductionHelperJIT().

---

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

• /home/jenkins-slave/workspace/core-os-doxygen/QueryEngine/ResultSetReductionJIT.h
• /home/jenkins-slave/workspace/core-os-doxygen/QueryEngine/ResultSetReductionJIT.cpp