QueryCompilationDescriptor.cpp

Go to the documentation of this file.

/*
 * Copyright 2022 HEAVY.AI, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "QueryCompilationDescriptor.h"

#include "QueryEngine/Execute.h"

std::unique_ptr<QueryMemoryDescriptor> QueryCompilationDescriptor::compile(
    const size_t max_groups_buffer_entry_guess,
    const int8_t crt_min_byte_width,
    const bool has_cardinality_estimation,
    const RelAlgExecutionUnit& ra_exe_unit,
    const std::vector<InputTableInfo>& table_infos,
    const PlanState::DeletedColumnsMap& deleted_cols_map,
    const ColumnFetcher& column_fetcher,
    const CompilationOptions& co,
    const ExecutionOptions& eo,
    RenderInfo* render_info,
    Executor* executor) {
  compilation_device_type_ = co.device_type;
  hoist_literals_ = co.hoist_literals;
  CHECK(executor);
  std::unique_ptr<QueryMemoryDescriptor> query_mem_desc;
  const auto data_mgr = executor->getDataMgr();
  try {
    std::tie(compilation_result_, query_mem_desc) = executor->compileWorkUnit(
        table_infos,
        deleted_cols_map,
        ra_exe_unit,
        co,
        eo,
        data_mgr->getCudaMgr(),
        g_enable_lazy_fetch &&
            co.allow_lazy_fetch,  // TODO(adb): remove param and just read from CO
        executor->row_set_mem_owner_,
        max_groups_buffer_entry_guess,
        crt_min_byte_width,
        has_cardinality_estimation,
        column_fetcher.columnarized_table_cache_,
        render_info);
  } catch (const CompilationRetryNoLazyFetch& e) {
    VLOG(1) << e.what();
    if (executor->cgen_state_->module_) {
      delete executor->cgen_state_->module_;
    }
    std::tie(compilation_result_, query_mem_desc) =
        executor->compileWorkUnit(table_infos,
                                  deleted_cols_map,
                                  ra_exe_unit,
                                  co,
                                  eo,
                                  data_mgr->getCudaMgr(),
                                  false,
                                  executor->row_set_mem_owner_,
                                  max_groups_buffer_entry_guess,
                                  crt_min_byte_width,
                                  has_cardinality_estimation,
                                  column_fetcher.columnarized_table_cache_,
                                  render_info);
  }
  actual_min_byte_width_ =
      std::max(query_mem_desc->updateActualMinByteWidth(MAX_BYTE_WIDTH_SUPPORTED),
               crt_min_byte_width);
  return query_mem_desc;
}