This is an automated email from the ASF dual-hosted git repository. syfeng pushed a commit to branch refactor-s4 in repository https://gitbox.apache.org/repos/asf/tvm.git
commit f187e1081508e138f3a0acaf4778960c9e5e8907 Author: Siyuan Feng <[email protected]> AuthorDate: Mon Feb 17 15:33:37 2025 +0800 fix --- docs/reference/api/python/contrib.rst | 6 ++---- include/tvm/driver/driver_api.h | 12 ------------ python/gen_requirements.py | 7 ------- src/target/codegen.cc | 2 -- src/te/operation/graph.cc | 1 + src/te/operation/graph.h | 6 +++--- 6 files changed, 6 insertions(+), 28 deletions(-) diff --git a/docs/reference/api/python/contrib.rst b/docs/reference/api/python/contrib.rst index e85d3bec5c..d0cf12b91c 100644 --- a/docs/reference/api/python/contrib.rst +++ b/docs/reference/api/python/contrib.rst @@ -48,20 +48,18 @@ tvm.contrib.dlpack .. automodule:: tvm.contrib.dlpack :members: + tvm.contrib.emcc ~~~~~~~~~~~~~~~~ .. automodule:: tvm.contrib.emcc :members: + tvm.contrib.miopen ~~~~~~~~~~~~~~~~~~ .. automodule:: tvm.contrib.miopen :members: -tvm.contrib.mxnet -~~~~~~~~~~~~~~~~~ -.. automodule:: tvm.contrib.mxnet - :members: tvm.contrib.ndk ~~~~~~~~~~~~~~~ diff --git a/include/tvm/driver/driver_api.h b/include/tvm/driver/driver_api.h index eaf737088b..39444d1629 100644 --- a/include/tvm/driver/driver_api.h +++ b/include/tvm/driver/driver_api.h @@ -88,18 +88,6 @@ TVM_DLL IRModule LowerModule(IRModule mod, bool simple_mode = false); TVM_DLL IRModule LowerPrimFunc(tvm::tir::PrimFunc func, const std::string& name, bool simple_mode = false); -/*! - * \brief Build an IRModule given a TE schedule, args and binds. This function also applies - * the lowering passes defined in CreatePassList. - * \param sch The TE schedule to lower. - * \param args The arguments to the function. - * \param name The name of the lowered function. - * \param binds Buffer assignments. - * \param global_var_supply The GlobalVarSupply to be used in the module. - * \param simple_mode Disables the loop partition pass. Defaults to false. - * \return The result module. - */ - /*! * \brief Build a device and host module for a specific target from an IRModule. * \param funcs The functions to be built. diff --git a/python/gen_requirements.py b/python/gen_requirements.py index dc776324bf..b5d8a56657 100644 --- a/python/gen_requirements.py +++ b/python/gen_requirements.py @@ -197,24 +197,17 @@ CONSTRAINTS = [ "docutils", "<0.17", ), # Work around https://github.com/readthedocs/sphinx_rtd_theme/issues/1115 - ("ethos-u-vela", "==3.8.0"), ("future", None), - ("h5py", "==2.10.0"), ("image", None), ("matplotlib", None), ("numpy", None), ("onnx", None), ("onnxoptimizer", None), ("onnxruntime", None), - ("opencv-python", None), ("pillow", None), - ("progressbar", None), - ("protobuf", None), ("psutil", None), ("pylint", None), - ("scikit-image", None), ("scipy", None), - ("six", None), ("sphinx", None), ("sphinx_autodoc_annotation", None), ("sphinx_gallery", None), diff --git a/src/target/codegen.cc b/src/target/codegen.cc index e5e2c4297c..b2fc88ed82 100644 --- a/src/target/codegen.cc +++ b/src/target/codegen.cc @@ -38,12 +38,10 @@ #include <vector> #include "../runtime/library_module.h" -#include "../support/base64.h" namespace tvm { namespace codegen { - runtime::Module Build(IRModule mod, Target target) { if (transform::PassContext::Current() ->GetConfig<Bool>("tir.disable_assert", Bool(false)) diff --git a/src/te/operation/graph.cc b/src/te/operation/graph.cc index cddace2a82..aee7f2afb1 100644 --- a/src/te/operation/graph.cc +++ b/src/te/operation/graph.cc @@ -29,6 +29,7 @@ #include <tvm/tir/stmt_functor.h> #include <unordered_set> +#include <vector> namespace tvm { namespace te { diff --git a/src/te/operation/graph.h b/src/te/operation/graph.h index fbb1241ad5..51ab8e1aa7 100644 --- a/src/te/operation/graph.h +++ b/src/te/operation/graph.h @@ -21,8 +21,8 @@ * \file graph.h * \brief Utilities to get information about schedule graph. */ -#ifndef TVM_TE_SCHEDULE_GRAPH_H_ -#define TVM_TE_SCHEDULE_GRAPH_H_ +#ifndef TVM_TE_OPERATION_GRAPH_H_ +#define TVM_TE_OPERATION_GRAPH_H_ #include <tvm/te/operation.h> #include <tvm/tir/expr.h> @@ -59,4 +59,4 @@ Array<Operation> PostDFSOrder(const Array<Operation>& roots, const ReadGraph& g) } // namespace te } // namespace tvm -#endif // TVM_TE_SCHEDULE_GRAPH_H_ +#endif // TVM_TE_OPERATION_GRAPH_H_
