67 строки
2.1 KiB
C++
67 строки
2.1 KiB
C++
/*
|
|
* Licensed to the Apache Software Foundation (ASF) under one
|
|
* or more contributor license agreements. See the NOTICE file
|
|
* distributed with this work for additional information
|
|
* regarding copyright ownership. The ASF licenses this file
|
|
* to you 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.
|
|
*/
|
|
|
|
/*!
|
|
* \file tvm/codegen.h
|
|
* \brief Collection of Lowlevel IR pass to codegen.
|
|
*/
|
|
#ifndef TVM_CODEGEN_H_
|
|
#define TVM_CODEGEN_H_
|
|
|
|
#include <string>
|
|
#include "base.h"
|
|
#include "expr.h"
|
|
#include "lowered_func.h"
|
|
#include "api_registry.h"
|
|
#include "runtime/packed_func.h"
|
|
|
|
namespace tvm {
|
|
/*! \brief namespace for lowlevel IR pass and codegen */
|
|
namespace codegen {
|
|
// use packed function from runtime.
|
|
using runtime::PackedFunc;
|
|
using runtime::TVMArgs;
|
|
using runtime::TVMRetValue;
|
|
|
|
/*!
|
|
* \brief Build a module from array of lowered function.
|
|
* \param funcs The functions to be built.
|
|
* \param target The target to be built.
|
|
* \return The builded module.
|
|
*
|
|
* \note Calls global API function "_codegen_build_" + target
|
|
*/
|
|
runtime::Module Build(const Array<LoweredFunc>& funcs,
|
|
const std::string& target);
|
|
/*!
|
|
* \brief Pack imported device library to a C file.
|
|
* Compile the C file and link with the host library
|
|
* will allow the DSO loader to automatically discover and import
|
|
* the dependency from the shared library.
|
|
*
|
|
* \param m The host module with the imports.
|
|
* \param system_lib Whether expose as system library.
|
|
* \return cstr The C string representation of the file.
|
|
*/
|
|
std::string PackImportsToC(const runtime::Module& m, bool system_lib);
|
|
} // namespace codegen
|
|
} // namespace tvm
|
|
|
|
#endif // TVM_CODEGEN_H_
|