Unverified Commit fc1a1d83 by Tianqi Chen Committed by GitHub

[REFACTOR] driver.h -> driver_api.h (#4760)

"driver" normally refers to the "main" function.
Rationale: the header exposes set of APIs to drive compilation
and should be named as driver api to best reflect its usage.
parent dcb556da
...@@ -18,16 +18,16 @@ ...@@ -18,16 +18,16 @@
*/ */
/*! /*!
* \file tvm/driver/driver.h * \file tvm/driver/driver_api.h
* \brief Compiler driver utilities. * \brief Compiler driver APIs to drive the compilation.
* *
* This module provides end-to-end utils to drive the compilation process. * This module provides end-to-end utils to drive the compilation process.
* We adopt the term "compiler driver" in common compiler infrastructures. * We adopt the term "compiler driver" in common compiler infrastructures.
* Note that a compiler driver is different from "runtime drivers". * Note that a compiler driver is different from "runtime drivers".
* Most of runtime related code are defined in the runtime folder instead. * Most of runtime related code are defined in the runtime folder instead.
*/ */
#ifndef TVM_DRIVER_DRIVER_H_ #ifndef TVM_DRIVER_DRIVER_API_H_
#define TVM_DRIVER_DRIVER_H_ #define TVM_DRIVER_DRIVER_API_H_
#include <tvm/runtime/packed_func.h> #include <tvm/runtime/packed_func.h>
#include <tvm/target/target.h> #include <tvm/target/target.h>
...@@ -115,4 +115,4 @@ TVM_DLL runtime::Module build(const Map<std::string, Array<tir::LoweredFunc>>& i ...@@ -115,4 +115,4 @@ TVM_DLL runtime::Module build(const Map<std::string, Array<tir::LoweredFunc>>& i
const BuildConfig& config); const BuildConfig& config);
} // namespace tvm } // namespace tvm
#endif // TVM_DRIVER_DRIVER_H_ #endif // TVM_DRIVER_DRIVER_API_H_
...@@ -26,9 +26,10 @@ There can be internal header files within each module that sit in src. ...@@ -26,9 +26,10 @@ There can be internal header files within each module that sit in src.
- node: base infra for IR/AST nodes that is dialect independent. - node: base infra for IR/AST nodes that is dialect independent.
- ir: Common IR infrastructure. - ir: Common IR infrastructure.
- tir: Tensor-level IR. - tir: Tensor-level IR.
- te: tensor expression DSL
- arith: Arithmetic expression and set simplification. - arith: Arithmetic expression and set simplification.
- top: tensor operation DSL for compute and schedule.
- relay: Relay IR, high-level optimization. - relay: Relay IR, high-level optimization.
- autotvm: The auto-tuning module. - autotvm: The auto-tuning module.
- contrib: Contrib extension libraries. - contrib: Contrib extension libraries.
- api: API function registration. - api: API function registration.
- driver: Compilation driver APIs.
...@@ -29,7 +29,7 @@ ...@@ -29,7 +29,7 @@
#include <tvm/te/schedule.h> #include <tvm/te/schedule.h>
#include <tvm/runtime/registry.h> #include <tvm/runtime/registry.h>
#include <tvm/driver/driver.h> #include <tvm/driver/driver_api.h>
#include <tvm/tir/data_layout.h> #include <tvm/tir/data_layout.h>
......
...@@ -19,10 +19,10 @@ ...@@ -19,10 +19,10 @@
/*! /*!
* Compile executable modules. * Compile executable modules.
* \file driver.cc * \file driver_api.cc
*/ */
#include <dmlc/thread_local.h> #include <dmlc/thread_local.h>
#include <tvm/driver/driver.h> #include <tvm/driver/driver_api.h>
#include <tvm/te/operation.h> #include <tvm/te/operation.h>
#include <tvm/tir/ir_pass.h> #include <tvm/tir/ir_pass.h>
#include <tvm/target/codegen.h> #include <tvm/target/codegen.h>
......
...@@ -22,7 +22,7 @@ ...@@ -22,7 +22,7 @@
* \brief Code generation for TVM's graph runtime. * \brief Code generation for TVM's graph runtime.
*/ */
#include <tvm/relay/analysis.h> #include <tvm/relay/analysis.h>
#include <tvm/driver/driver.h> #include <tvm/driver/driver_api.h>
#include <tvm/runtime/device_api.h> #include <tvm/runtime/device_api.h>
#include <tvm/runtime/vm.h> #include <tvm/runtime/vm.h>
#include <tvm/relay/expr.h> #include <tvm/relay/expr.h>
......
...@@ -32,7 +32,7 @@ ...@@ -32,7 +32,7 @@
#include <tvm/relay/expr_functor.h> #include <tvm/relay/expr_functor.h>
#include <tvm/relay/op.h> #include <tvm/relay/op.h>
#include <tvm/relay/op_attr_types.h> #include <tvm/relay/op_attr_types.h>
#include <tvm/driver/driver.h> #include <tvm/driver/driver_api.h>
#include <topi/tags.h> #include <topi/tags.h>
#include <utility> #include <utility>
......
...@@ -30,7 +30,7 @@ ...@@ -30,7 +30,7 @@
#include <tvm/relay/analysis.h> #include <tvm/relay/analysis.h>
#include <tvm/relay/attrs/debug.h> #include <tvm/relay/attrs/debug.h>
#include <tvm/relay/feature.h> #include <tvm/relay/feature.h>
#include <tvm/driver/driver.h> #include <tvm/driver/driver_api.h>
#include "compile_engine.h" #include "compile_engine.h"
......
...@@ -27,7 +27,7 @@ ...@@ -27,7 +27,7 @@
#include <dmlc/json.h> #include <dmlc/json.h>
#include <tvm/relay/expr.h> #include <tvm/relay/expr.h>
#include <tvm/relay/type.h> #include <tvm/relay/type.h>
#include <tvm/driver/driver.h> #include <tvm/driver/driver_api.h>
#include <tvm/target/codegen.h> #include <tvm/target/codegen.h>
#include <tvm/tir/ir_pass.h> #include <tvm/tir/ir_pass.h>
#include <tvm/te/operation.h> #include <tvm/te/operation.h>
......
...@@ -31,7 +31,7 @@ ...@@ -31,7 +31,7 @@
#include <tvm/relay/transform.h> #include <tvm/relay/transform.h>
#include <tvm/runtime/vm.h> #include <tvm/runtime/vm.h>
#include <tvm/relay/attrs/memory.h> #include <tvm/relay/attrs/memory.h>
#include <tvm/driver/driver.h> #include <tvm/driver/driver_api.h>
#include <iostream> #include <iostream>
#include <memory> #include <memory>
......
...@@ -22,7 +22,7 @@ ...@@ -22,7 +22,7 @@
#include <topi/cuda/injective.h> #include <topi/cuda/injective.h>
#include <tvm/te/operation.h> #include <tvm/te/operation.h>
#include <tvm/runtime/registry.h> #include <tvm/runtime/registry.h>
#include <tvm/driver/driver.h> #include <tvm/driver/driver_api.h>
#include <string> #include <string>
#include <cmath> #include <cmath>
......
...@@ -18,7 +18,7 @@ ...@@ -18,7 +18,7 @@
*/ */
#include <gtest/gtest.h> #include <gtest/gtest.h>
#include <tvm/driver/driver.h> #include <tvm/driver/driver_api.h>
#include <tvm/te/operation.h> #include <tvm/te/operation.h>
#include <tvm/relay/expr.h> #include <tvm/relay/expr.h>
#include <tvm/relay/type.h> #include <tvm/relay/type.h>
......
...@@ -19,7 +19,7 @@ ...@@ -19,7 +19,7 @@
#include <gtest/gtest.h> #include <gtest/gtest.h>
#include <topi/generic/injective.h> #include <topi/generic/injective.h>
#include <tvm/driver/driver.h> #include <tvm/driver/driver_api.h>
#include <tvm/relay/expr.h> #include <tvm/relay/expr.h>
#include <tvm/ir/module.h> #include <tvm/ir/module.h>
#include <tvm/relay/analysis.h> #include <tvm/relay/analysis.h>
......
...@@ -31,7 +31,7 @@ ...@@ -31,7 +31,7 @@
#include <gtest/gtest.h> #include <gtest/gtest.h>
#include <topi/generic/injective.h> #include <topi/generic/injective.h>
#include <tvm/driver/driver.h> #include <tvm/driver/driver_api.h>
#include <tvm/te/operation.h> #include <tvm/te/operation.h>
#include <tvm/relay/analysis.h> #include <tvm/relay/analysis.h>
#include <tvm/relay/expr.h> #include <tvm/relay/expr.h>
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment