SpeculativeTopN.cpp File Reference

#include "SpeculativeTopN.h"
#include "Logger/Logger.h"
#include "RelAlgExecutor.h"
#include "ResultSet.h"

Include dependency graph for SpeculativeTopN.cpp:

Go to the source code of this file.

Functions
bool	use_speculative_top_n (const RelAlgExecutionUnit &ra_exe_unit, const QueryMemoryDescriptor &query_mem_desc)

Function Documentation

bool use_speculative_top_n	(	const RelAlgExecutionUnit &	ra_exe_unit,
		const QueryMemoryDescriptor &	query_mem_desc
	)

SpeculativeTopN sort is used when there are multiple already sorted results (when GPU sort is used on multiple devices, refer to GroupByAndAggregate::gpuCanHandleOrderEntries), and we want to pick top n elements (LIMIT caluse exists), and we have already chosen this algorithm when creating the proper work unit (refer to RelAlgExecutor::createSortInputWorkUnit).

Besides, we currently only support cases with 2 target expressions and only with COUNT aggregate (similar limitations exists in whether or not we support GPU sort).

Definition at line 188 of file SpeculativeTopN.cpp.

References SortInfo::algorithm, g_cluster, SortInfo::limit, RelAlgExecutionUnit::sort_info, QueryMemoryDescriptor::sortOnGpu(), SpeculativeTopN, and RelAlgExecutionUnit::target_exprs.

Referenced by Executor::collectAllDeviceResults(), RelAlgExecutor::executeSort(), RelAlgExecutor::executeWorkUnit(), and QueryExecutionContext::launchGpuCode().

                                                                        {
  if (g_cluster) {
    return false;
  }
  if (ra_exe_unit.target_exprs.size() != 2) {
    return false;
  }
  for (const auto target_expr : ra_exe_unit.target_exprs) {
    const auto agg_expr = dynamic_cast<const Analyzer::AggExpr*>(target_expr);
    if (agg_expr && !shared::is_any<kCOUNT, kCOUNT_IF>(agg_expr->get_aggtype())) {
      return false;
    }
  }
  return query_mem_desc.sortOnGpu() && ra_exe_unit.sort_info.limit &&
         ra_exe_unit.sort_info.algorithm == SortAlgorithm::SpeculativeTopN;
}

Here is the call graph for this function:

Here is the caller graph for this function: