You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by tq...@apache.org on 2025/02/17 16:01:19 UTC
(tvm) 08/09: fix
This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a commit to branch refactor
in repository https://gitbox.apache.org/repos/asf/tvm.git
commit 3fed092f47c42f2abb34b68349a01c076d21949c
Author: Siyuan Feng <hz...@sjtu.edu.cn>
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_