wehub-resource-sync
95 lines
3.4 KiB
C++
95 lines
3.4 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/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_
|