anonymous_namespace{TableFunctionExecutionContext.cpp} Namespace Reference

Enumerations
enum	{   ERROR_BUFFER, COL_BUFFERS, COL_SIZES, OUTPUT_BUFFERS,   OUTPUT_ROW_COUNT, KERNEL_PARAM_COUNT }

Functions
template<typename T >
const int8_t *	create_literal_buffer (T literal, const ExecutorDeviceType device_type, std::vector< std::unique_ptr< char[]>> &literals_owner, CudaAllocator *gpu_allocator)
size_t	get_output_row_count (const TableFunctionExecutionUnit &exe_unit, size_t input_element_count)

Enumeration Type Documentation

anonymous enum

Enumerator
ERROR_BUFFER
COL_BUFFERS
COL_SIZES
OUTPUT_BUFFERS
OUTPUT_ROW_COUNT
KERNEL_PARAM_COUNT

Definition at line 313 of file TableFunctionExecutionContext.cpp.

     {
  ERROR_BUFFER,
  COL_BUFFERS,
  COL_SIZES,
  OUTPUT_BUFFERS,
  OUTPUT_ROW_COUNT,
  KERNEL_PARAM_COUNT,
};

Function Documentation

template<typename T >

const int8_t* anonymous_namespace{TableFunctionExecutionContext.cpp}::create_literal_buffer	(	T	literal,
		const ExecutorDeviceType	device_type,
		std::vector< std::unique_ptr< char[]>> &	literals_owner,
		CudaAllocator *	gpu_allocator
	)

Definition at line 28 of file TableFunctionExecutionContext.cpp.

References CudaAllocator::alloc(), CHECK, CHECK_LE, CudaAllocator::copyToDevice(), CPU, GPU, omnisci.dtypes::T, and UNREACHABLE.

Referenced by TableFunctionExecutionContext::execute().

                                                                  {
  CHECK_LE(sizeof(T), sizeof(int64_t));  // pad to 8 bytes
  switch (device_type) {
    case ExecutorDeviceType::CPU: {
      literals_owner.emplace_back(std::make_unique<char[]>(sizeof(int64_t)));
      std::memcpy(literals_owner.back().get(), &literal, sizeof(T));
      return reinterpret_cast<const int8_t*>(literals_owner.back().get());
    }
    case ExecutorDeviceType::GPU: {
      CHECK(gpu_allocator);
      const auto gpu_literal_buf_ptr = gpu_allocator->alloc(sizeof(int64_t));
      gpu_allocator->copyToDevice(
          gpu_literal_buf_ptr, reinterpret_cast<int8_t*>(&literal), sizeof(T));
      return gpu_literal_buf_ptr;
    }
  }
  UNREACHABLE();
  return nullptr;
}

Here is the call graph for this function:

Here is the caller graph for this function:

size_t anonymous_namespace{TableFunctionExecutionContext.cpp}::get_output_row_count	(	const TableFunctionExecutionUnit &	exe_unit,
		size_t	input_element_count
	)

Definition at line 51 of file TableFunctionExecutionContext.cpp.

References table_functions::TableFunction::getOutputRowSizeType(), table_functions::kConstant, table_functions::kUserSpecifiedConstantParameter, table_functions::kUserSpecifiedRowMultiplier, TableFunctionExecutionUnit::output_buffer_size_param, TableFunctionExecutionUnit::table_func, and UNREACHABLE.

Referenced by TableFunctionExecutionContext::launchCpuCode(), and TableFunctionExecutionContext::launchGpuCode().

                                                        {
  size_t allocated_output_row_count = 0;
  switch (exe_unit.table_func.getOutputRowSizeType()) {
    case table_functions::OutputBufferSizeType::kConstant:
    case table_functions::OutputBufferSizeType::kUserSpecifiedConstantParameter: {
      allocated_output_row_count = exe_unit.output_buffer_size_param;
      break;
    }
    case table_functions::OutputBufferSizeType::kUserSpecifiedRowMultiplier: {
      allocated_output_row_count =
          exe_unit.output_buffer_size_param * input_element_count;
      break;
    }
    default: {
      UNREACHABLE();
    }
  }
  return allocated_output_row_count;
}

Here is the call graph for this function:

Here is the caller graph for this function: