TreeModelPredictionMgr.cpp File Reference

#include "TreeModelPredictionMgr.h"
#include "CodeGenerator.h"
#include "Parser/ParserNode.h"
#include <tbb/parallel_for.h>

Include dependency graph for TreeModelPredictionMgr.cpp:

Go to the source code of this file.

Functions
std::pair< std::vector < std::shared_ptr< const Analyzer::Constant > >, std::vector< const Analyzer::Constant * > >	generate_kernel_buffer_constants (CgenState *cgen_state_ptr, const std::vector< const int8_t * > &kernel_buffers, const bool hoist_literals)

Function Documentation

std::pair<std::vector<std::shared_ptr<const Analyzer::Constant> >, std::vector<const Analyzer::Constant*> > generate_kernel_buffer_constants	(	CgenState *	cgen_state_ptr,
		const std::vector< const int8_t * > &	kernel_buffers,
		const bool	hoist_literals
	)

Definition at line 137 of file TreeModelPredictionMgr.cpp.

References Parser::IntLiteral::analyzeValue(), CHECK, CHECK_EQ, CHECK_GE, and kENCODING_NONE.

Referenced by TreeModelPredictionMgr::codegen().

                                                            {
  std::vector<std::shared_ptr<const Analyzer::Constant>> kernel_buffer_constants_owned;
  std::vector<const Analyzer::Constant*> kernel_buffer_constants;
  for (const auto kernel_buffer : kernel_buffers) {
    const int64_t kernel_buffer_handle = reinterpret_cast<int64_t>(kernel_buffer);
    const auto kernel_buffer_handle_literal =
        std::dynamic_pointer_cast<Analyzer::Constant>(
            Parser::IntLiteral::analyzeValue(kernel_buffer_handle));
    CHECK_EQ(kENCODING_NONE,
             kernel_buffer_handle_literal->get_type_info().get_compression());
    kernel_buffer_constants_owned.push_back(kernel_buffer_handle_literal);
    kernel_buffer_constants.push_back(kernel_buffer_handle_literal.get());
  }
  CHECK_GE(kernel_buffer_constants.size(), 1UL);
  CHECK(hoist_literals || kernel_buffer_constants.size() == 1UL);

  return std::make_pair(kernel_buffer_constants_owned, kernel_buffer_constants);
}

Here is the call graph for this function:

Here is the caller graph for this function: