chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,94 @@
|
||||
/*
|
||||
* 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/target/codegen.h
|
||||
* \brief Translates IRModule to runtime::Module.
|
||||
*/
|
||||
#ifndef TVM_TARGET_CODEGEN_H_
|
||||
#define TVM_TARGET_CODEGEN_H_
|
||||
|
||||
#include <tvm/ffi/extra/module.h>
|
||||
#include <tvm/ir/module.h>
|
||||
#include <tvm/target/target.h>
|
||||
#include <tvm/tirx/expr.h>
|
||||
|
||||
#include <string>
|
||||
|
||||
namespace tvm {
|
||||
/*! \brief namespace for target translation and codegen. */
|
||||
namespace codegen {
|
||||
|
||||
/*!
|
||||
* \brief Build a module from array of lowered function.
|
||||
* \param mod The Module to be built
|
||||
* \param target The target to be built.
|
||||
* \return The result runtime::Module.
|
||||
*/
|
||||
ffi::Module Build(IRModule mod, Target target);
|
||||
|
||||
/*!
|
||||
* \brief Serialize runtime module including its submodules
|
||||
* \param mod The runtime module to serialize including its import tree.
|
||||
* \param export_dso By default, include the info of DSOExportable modules. If disabled, an error
|
||||
* will be raised when encountering DSO modules.
|
||||
*/
|
||||
std::string SerializeModuleToBytes(const ffi::Module& mod, bool export_dso = true);
|
||||
|
||||
/*!
|
||||
* \brief Deserialize runtime module including its submodules
|
||||
* \param blob byte stream, which are generated by `SerializeModuleToBytes`.
|
||||
* \return runtime::Module runtime module constructed from the given stream
|
||||
*/
|
||||
ffi::Module DeserializeModuleFromBytes(std::string blob);
|
||||
|
||||
/*!
|
||||
* \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.
|
||||
* \param c_symbol_prefix Optional symbol prefix of the blob symbol.
|
||||
* \return cstr The C string representation of the file.
|
||||
*/
|
||||
std::string PackImportsToC(const ffi::Module& m, bool system_lib,
|
||||
const std::string& c_symbol_prefix = "");
|
||||
|
||||
/*!
|
||||
* \brief Pack imported device library to a LLVM module.
|
||||
* Compile the LLVM module 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.
|
||||
* \param target_triple LLVM target triple
|
||||
* \param c_symbol_prefix Optional symbol prefix of the blob symbol.
|
||||
*
|
||||
* \return runtime::Module The generated LLVM module.
|
||||
*/
|
||||
ffi::Module PackImportsToLLVM(const ffi::Module& m, bool system_lib,
|
||||
const std::string& target_triple,
|
||||
const std::string& c_symbol_prefix = "");
|
||||
|
||||
} // namespace codegen
|
||||
} // namespace tvm
|
||||
#endif // TVM_TARGET_CODEGEN_H_
|
||||
Reference in New Issue
Block a user