/*! * Copyright (c) 2016 by Contributors * Implementation of API functions related to Codegen * \file c_api_codegen.cc */ #include <tvm/expr.h> #include <tvm/ir.h> #include <tvm/codegen.h> #include <tvm/lowered_func.h> #include <tvm/api_registry.h> namespace tvm { namespace codegen { TVM_REGISTER_API("codegen._Build") .set_body([](TVMArgs args, TVMRetValue *ret) { if (args[0].IsNodeType<LoweredFunc>()) { *ret = Build({args[0]}, args[1]); } else { *ret = Build(args[0], args[1]); } }); TVM_REGISTER_API("module._PackImportsToC") .set_body([](TVMArgs args, TVMRetValue *ret) { *ret = PackImportsToC(args[0], args[1]); }); } // namespace codegen } // namespace tvm