7 #include <dmlc/registry.h> 15 Compiler* Compiler::Create(
const std::string& name,
17 auto *e = ::dmlc::Registry< ::treelite::CompilerReg>::Get()->Find(name);
19 LOG(FATAL) <<
"Unknown compiler type " << name;
21 return (e->body)(param);
28 DMLC_REGISTER_PARAMETER(CompilerParam);
31 DMLC_REGISTRY_LINK_TAG(ast_native);
parameters for tree compiler
Parameters for tree compiler.
Interface of compiler that compiles a tree ensemble model.
Registry entry for compiler.