wehub/apache--tvm
1
0
Fork 0
Code Issues Pull Requests Actions Deployments Models Agent Notes Packages Projects Releases Wiki Activity

chore: import upstream snapshot with attribution
Lint / lint (push) Has been cancelled
Details
CI / MacOS (push) Has been cancelled
Details
CI / Windows (push) Has been cancelled
Details

Browse Source
This commit is contained in:
wehub-resource-sync
2026-07-13 13:36:25 +08:00
commit 26446540fa
3151 changed files with 974126 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/reference/api/links.rst
+26
View File
@@ -0,0 +1,26 @@
.. 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.
Other APIs
==========
This page contains links to API references that are built with different doc
build system.
* `C++ doxygen API <doxygen/index.html>`_
* `Typescript typedoc API <typedoc/index.html>`_
* `Java Javadoc API <javadoc/index.html>`_
docs/reference/api/python/arith.rst
+22
View File
@@ -0,0 +1,22 @@
.. 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.
tvm.arith
---------
.. automodule:: tvm.arith
:members:
:imported-members:
docs/reference/api/python/contrib.rst
+123
View File
@@ -0,0 +1,123 @@
.. 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.
tvm.contrib
-----------
.. automodule:: tvm.contrib
tvm.contrib.cblas
~~~~~~~~~~~~~~~~~
.. automodule:: tvm.contrib.cblas
:members:
tvm.contrib.coreml_runtime
~~~~~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: tvm.contrib.coreml_runtime
:members:
tvm.contrib.cublas
~~~~~~~~~~~~~~~~~~
.. automodule:: tvm.contrib.cublas
:members:
tvm.contrib.cublaslt
~~~~~~~~~~~~~~~~~~~~
.. automodule:: tvm.contrib.cublaslt
:members:
tvm.contrib.cudnn
~~~~~~~~~~~~~~~~~
.. automodule:: tvm.contrib.cudnn
:members:
tvm.contrib.dlpack
~~~~~~~~~~~~~~~~~~
.. automodule:: tvm.contrib.dlpack
:members:
tvm.contrib.dnnl
~~~~~~~~~~~~~~~~
.. automodule:: tvm.contrib.dnnl
:members:
tvm.contrib.download
~~~~~~~~~~~~~~~~~~~~
.. automodule:: tvm.contrib.download
:members:
tvm.contrib.hipblas
~~~~~~~~~~~~~~~~~~~
.. automodule:: tvm.contrib.hipblas
:members:
tvm.contrib.mkl
~~~~~~~~~~~~~~~
.. automodule:: tvm.contrib.mkl
:members:
tvm.contrib.nnpack
~~~~~~~~~~~~~~~~~~
.. automodule:: tvm.contrib.nnpack
:members:
tvm.contrib.pickle_memoize
~~~~~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: tvm.contrib.pickle_memoize
:members:
tvm.contrib.random
~~~~~~~~~~~~~~~~~~
.. automodule:: tvm.contrib.random
:members:
tvm.contrib.thrust
~~~~~~~~~~~~~~~~~~
.. automodule:: tvm.contrib.thrust
:members:
tvm.contrib.tvmjs
~~~~~~~~~~~~~~~~~
.. automodule:: tvm.contrib.tvmjs
:members:
tvm.contrib.cutlass
~~~~~~~~~~~~~~~~~~~
.. automodule:: tvm.contrib.cutlass
:members:
:imported-members:
tvm.contrib.hexagon
~~~~~~~~~~~~~~~~~~~
.. automodule:: tvm.contrib.hexagon
:members:
:imported-members:
docs/reference/api/python/driver.rst
+22
View File
@@ -0,0 +1,22 @@
.. 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.
tvm.driver
----------
.. automodule:: tvm.driver
.. autofunction:: tvm.compile
docs/reference/api/python/error.rst
+23
View File
@@ -0,0 +1,23 @@
.. 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.
tvm.error
---------
.. automodule:: tvm.error
:members:
:imported-members:
:autosummary:
docs/reference/api/python/exec.rst
+22
View File
@@ -0,0 +1,22 @@
.. 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.
tvm.exec
--------
.. automodule:: tvm.exec
:members:
:imported-members:
docs/reference/api/python/index.rst
+100
View File
@@ -0,0 +1,100 @@
.. 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.
Python API
==========
.. toctree::
:maxdepth: 1
:caption: tvm
error
ir
arith
instrument
transform
target
driver
testing
exec
support
.. toctree::
:maxdepth: 1
:caption: tvm.runtime
runtime/runtime
runtime/vm
runtime/disco
.. toctree::
:maxdepth: 1
:caption: tvm.relax
relax/relax
relax/analysis
relax/backend
relax/block_builder
relax/distributed
relax/frontend
relax/op
relax/testing
relax/training
relax/transform
relax/dpl
.. toctree::
:maxdepth: 1
:caption: tvm.s_tir
s_tir/analysis
s_tir/schedule
s_tir/transform
s_tir/dlight
s_tir/backend
s_tir/tensor_intrin
.. toctree::
:maxdepth: 1
:caption: tvm.script
script/script
script/parser
script/ir_builder
script/printer
.. toctree::
:maxdepth: 1
:caption: tvm.te
te
topi
.. toctree::
:maxdepth: 1
:caption: tvm.s_tir.meta_schedule
meta_schedule
.. toctree::
:maxdepth: 1
:caption: Misc
rpc
contrib
docs/reference/api/python/instrument.rst
+23
View File
@@ -0,0 +1,23 @@
.. 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.
tvm.ir.instrument
------------------
.. automodule:: tvm.ir.instrument
:members:
:imported-members:
:exclude-members: Path
docs/reference/api/python/ir.rst
+24
View File
@@ -0,0 +1,24 @@
.. 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.
tvm.ir
------
.. automodule:: tvm.ir
:members:
:imported-members:
:exclude-members: Expr
:autosummary:
docs/reference/api/python/meta_schedule.rst
+146
View File
@@ -0,0 +1,146 @@
.. 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.
tvm.s_tir.meta_schedule
------------------------
tvm.s_tir.meta_schedule
***********************
.. automodule:: tvm.s_tir.meta_schedule
:members:
:imported-members:
:autosummary:
tvm.s_tir.meta_schedule.arg_info
********************************
.. automodule:: tvm.s_tir.meta_schedule.arg_info
:members:
:noindex:
tvm.s_tir.meta_schedule.builder
*******************************
.. automodule:: tvm.s_tir.meta_schedule.builder
:members:
:imported-members:
:noindex:
tvm.s_tir.meta_schedule.cost_model
**********************************
.. automodule:: tvm.s_tir.meta_schedule.cost_model
:members:
:imported-members:
:noindex:
tvm.s_tir.meta_schedule.database
********************************
.. automodule:: tvm.s_tir.meta_schedule.database
:members:
:imported-members:
:noindex:
tvm.s_tir.meta_schedule.feature_extractor
*****************************************
.. automodule:: tvm.s_tir.meta_schedule.feature_extractor
:members:
:imported-members:
:noindex:
tvm.s_tir.meta_schedule.measure_callback
****************************************
.. automodule:: tvm.s_tir.meta_schedule.measure_callback
:members:
:imported-members:
:noindex:
tvm.s_tir.meta_schedule.mutator
*******************************
.. automodule:: tvm.s_tir.meta_schedule.mutator
:members:
:imported-members:
:noindex:
tvm.s_tir.meta_schedule.postproc
********************************
.. automodule:: tvm.s_tir.meta_schedule.postproc
:members:
:imported-members:
:noindex:
tvm.s_tir.meta_schedule.runner
******************************
.. automodule:: tvm.s_tir.meta_schedule.runner
:members:
:imported-members:
:noindex:
tvm.s_tir.meta_schedule.schedule_rule
*************************************
.. automodule:: tvm.s_tir.meta_schedule.schedule_rule
:members:
:noindex:
tvm.s_tir.meta_schedule.search_strategy
***************************************
.. automodule:: tvm.s_tir.meta_schedule.search_strategy
:members:
:imported-members:
:noindex:
tvm.s_tir.meta_schedule.space_generator
***************************************
.. automodule:: tvm.s_tir.meta_schedule.space_generator
:members:
:noindex:
tvm.s_tir.meta_schedule.task_scheduler
**************************************
.. automodule:: tvm.s_tir.meta_schedule.task_scheduler
:members:
:imported-members:
:noindex:
tvm.s_tir.meta_schedule.tir_integration
***************************************
.. automodule:: tvm.s_tir.meta_schedule.tir_integration
:members:
:noindex:
tvm.s_tir.meta_schedule.relax_integration
*****************************************
.. automodule:: tvm.s_tir.meta_schedule.relax_integration
:members:
:noindex:
tvm.s_tir.meta_schedule.trace_apply
************************************
.. automodule:: tvm.s_tir.meta_schedule.trace_apply
:members:
:noindex:
tvm.s_tir.meta_schedule.schedule
********************************
.. automodule:: tvm.s_tir.meta_schedule.schedule
:members:
:imported-members:
:noindex:
tvm.s_tir.meta_schedule.post_optimization
*****************************************
.. automodule:: tvm.s_tir.meta_schedule.post_optimization
:members:
:imported-members:
:noindex:
docs/reference/api/python/relax/analysis.rst
+22
View File
@@ -0,0 +1,22 @@
.. 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.
tvm.relax.analysis
------------------
.. automodule:: tvm.relax.analysis
:members:
:imported-members:
docs/reference/api/python/relax/backend.rst
+67
View File
@@ -0,0 +1,67 @@
.. 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.
tvm.relax.backend
-----------------
tvm.relax.backend
*****************
.. automodule:: tvm.relax.backend
:members:
:imported-members:
tvm.relax.backend.cuda
**********************
.. automodule:: tvm.relax.backend.cuda
:members:
:no-index:
tvm.relax.backend.rocm
**********************
.. automodule:: tvm.relax.backend.rocm
:members:
:no-index:
tvm.relax.backend.metal
***********************
.. automodule:: tvm.relax.backend.metal
:members:
:no-index:
tvm.relax.backend.adreno
************************
.. automodule:: tvm.relax.backend.adreno
:members:
:no-index:
tvm.relax.backend.gpu_generic
*****************************
.. automodule:: tvm.relax.backend.gpu_generic
:members:
:no-index:
tvm.relax.backend.cpu_generic
*****************************
.. automodule:: tvm.relax.backend.cpu_generic
:members:
:no-index:
tvm.relax.backend.contrib
*************************
.. automodule:: tvm.relax.backend.contrib
:members:
:no-index:
docs/reference/api/python/relax/block_builder.rst
+21
View File
@@ -0,0 +1,21 @@
.. 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.
tvm.relax.block_builder
-----------------------
.. automodule:: tvm.relax.block_builder
:members:
docs/reference/api/python/relax/distributed.rst
+28
View File
@@ -0,0 +1,28 @@
.. 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.
tvm.relax.distributed
---------------------
.. automodule:: tvm.relax.distributed
:members:
:imported-members:
tvm.relax.distributed.transform
*******************************
.. automodule:: tvm.relax.distributed.transform
:members:
:no-index:
docs/reference/api/python/relax/dpl.rst
+21
View File
@@ -0,0 +1,21 @@
.. 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.
tvm.relax.dpl
-------------
.. automodule:: tvm.relax.dpl
:members:
docs/reference/api/python/relax/frontend.rst
+60
View File
@@ -0,0 +1,60 @@
.. 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.
tvm.relax.frontend
------------------
.. automodule:: tvm.relax.frontend
:members:
:imported-members:
tvm.relax.frontend.nn
*********************
.. automodule:: tvm.relax.frontend.nn
:members:
:imported-members:
:exclude-members: BlockBuilder
:noindex:
tvm.relax.frontend.nn.llm
*************************
.. automodule:: tvm.relax.frontend.nn.llm
:members:
:no-index:
tvm.relax.frontend.onnx
***********************
.. automodule:: tvm.relax.frontend.onnx
:members:
:imported-members:
tvm.relax.frontend.stablehlo
****************************
.. automodule:: tvm.relax.frontend.stablehlo
:members:
:imported-members:
tvm.relax.frontend.tflite
*************************
.. automodule:: tvm.relax.frontend.tflite
:members:
:imported-members:
tvm.relax.frontend.torch
************************
.. automodule:: tvm.relax.frontend.torch
:members:
:imported-members:
docs/reference/api/python/relax/op.rst
+95
View File
@@ -0,0 +1,95 @@
.. 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.
tvm.relax.op
------------
tvm.relax.op
************
.. automodule:: tvm.relax.op
:members:
:imported-members:
:exclude-members: Expr
tvm.relax.op.nn
***************
.. automodule:: tvm.relax.op.nn
:members:
:imported-members:
:exclude-members: Expr
tvm.relax.op.builtin
********************
.. automodule:: tvm.relax.op.builtin
:members:
:imported-members:
:exclude-members: Expr
tvm.relax.op.ccl
****************
.. automodule:: tvm.relax.op.ccl
:members:
:imported-members:
:exclude-members: Expr
tvm.relax.op.distributed
************************
.. automodule:: tvm.relax.op.distributed
:members:
:imported-members:
:exclude-members: Expr
tvm.relax.op.grad
*****************
.. automodule:: tvm.relax.op.grad
:members:
:imported-members:
:exclude-members: Expr
tvm.relax.op.image
******************
.. automodule:: tvm.relax.op.image
:members:
:imported-members:
:exclude-members: Expr
tvm.relax.op.memory
*******************
.. automodule:: tvm.relax.op.memory
:members:
:imported-members:
:exclude-members: Expr
tvm.relax.op.vision
*******************
.. automodule:: tvm.relax.op.vision
:members:
:imported-members:
:exclude-members: Expr
tvm.relax.op.vm
***************
.. automodule:: tvm.relax.op.vm
:members:
:imported-members:
:exclude-members: Expr
tvm.relax.op.op_attrs
*********************
.. automodule:: tvm.relax.op.op_attrs
:members:
:exclude-members: Attrs
docs/reference/api/python/relax/relax.rst
+23
View File
@@ -0,0 +1,23 @@
.. 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.
tvm.relax
---------
.. automodule:: tvm.relax
:members:
:imported-members:
:exclude-members: BlockBuilder, Call, Span, GlobalVar, SourceName, TupleType, Type, FuncType
docs/reference/api/python/relax/testing.rst
+22
View File
@@ -0,0 +1,22 @@
.. 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.
tvm.relax.testing
-----------------
.. automodule:: tvm.relax.testing
:members:
:no-index:
docs/reference/api/python/relax/training.rst
+22
View File
@@ -0,0 +1,22 @@
.. 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.
tvm.relax.training
------------------
.. automodule:: tvm.relax.training
:members:
:imported-members:
docs/reference/api/python/relax/transform.rst
+131
View File
@@ -0,0 +1,131 @@
.. 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.
.. _api-relax-transformation:
tvm.relax.transform
-------------------
.. automodule:: tvm.relax.transform
:members:
:imported-members:
tvm.relax.transform.legalize_ops
*********************************
.. automodule:: tvm.relax.transform.legalize_ops
:members:
tvm.relax.transform.legalize_ops.binary
========================================
.. automodule:: tvm.relax.transform.legalize_ops.binary
:members:
:no-index:
tvm.relax.transform.legalize_ops.ccl
=====================================
.. automodule:: tvm.relax.transform.legalize_ops.ccl
:members:
:no-index:
tvm.relax.transform.legalize_ops.create
========================================
.. automodule:: tvm.relax.transform.legalize_ops.create
:members:
:no-index:
tvm.relax.transform.legalize_ops.datatype
==========================================
.. automodule:: tvm.relax.transform.legalize_ops.datatype
:members:
:no-index:
tvm.relax.transform.legalize_ops.distributed
==============================================
.. automodule:: tvm.relax.transform.legalize_ops.distributed
:members:
:no-index:
tvm.relax.transform.legalize_ops.grad
======================================
.. automodule:: tvm.relax.transform.legalize_ops.grad
:members:
:no-index:
tvm.relax.transform.legalize_ops.image
=======================================
.. automodule:: tvm.relax.transform.legalize_ops.image
:members:
:no-index:
tvm.relax.transform.legalize_ops.index
=======================================
.. automodule:: tvm.relax.transform.legalize_ops.index
:members:
:no-index:
tvm.relax.transform.legalize_ops.linear_algebra
================================================
.. automodule:: tvm.relax.transform.legalize_ops.linear_algebra
:members:
:no-index:
tvm.relax.transform.legalize_ops.manipulate
=============================================
.. automodule:: tvm.relax.transform.legalize_ops.manipulate
:members:
:no-index:
tvm.relax.transform.legalize_ops.nn
====================================
.. automodule:: tvm.relax.transform.legalize_ops.nn
:members:
:no-index:
tvm.relax.transform.legalize_ops.qdq
=====================================
.. automodule:: tvm.relax.transform.legalize_ops.qdq
:members:
:no-index:
tvm.relax.transform.legalize_ops.search
========================================
.. automodule:: tvm.relax.transform.legalize_ops.search
:members:
:no-index:
tvm.relax.transform.legalize_ops.statistical
==============================================
.. automodule:: tvm.relax.transform.legalize_ops.statistical
:members:
:no-index:
tvm.relax.transform.legalize_ops.unary
=======================================
.. automodule:: tvm.relax.transform.legalize_ops.unary
:members:
:no-index:
tvm.relax.transform.legalize_ops.vision
========================================
.. automodule:: tvm.relax.transform.legalize_ops.vision
:members:
:no-index:
tvm.relax.transform.legalize_ops.adreno
========================================
.. automodule:: tvm.relax.transform.legalize_ops.adreno
:members:
:no-index:
docs/reference/api/python/rpc.rst
+23
View File
@@ -0,0 +1,23 @@
.. 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.
tvm.rpc
-------
.. automodule:: tvm.rpc
:members:
:imported-members:
:autosummary:
docs/reference/api/python/runtime/disco.rst
+22
View File
@@ -0,0 +1,22 @@
.. 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.
tvm.runtime.disco
-----------------
.. automodule:: tvm.runtime.disco
:members:
:imported-members:
docs/reference/api/python/runtime/runtime.rst
+21
View File
@@ -0,0 +1,21 @@
.. 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.
tvm.runtime
-----------
.. automodule:: tvm.runtime
:members:
docs/reference/api/python/runtime/vm.rst
+21
View File
@@ -0,0 +1,21 @@
.. 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.
tvm.runtime.vm
--------------------
.. automodule:: tvm.runtime.vm
:members:
docs/reference/api/python/s_tir/analysis.rst
+21
View File
@@ -0,0 +1,21 @@
.. 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.
tvm.s_tir.analysis
------------------
.. automodule:: tvm.s_tir.analysis
:members:
docs/reference/api/python/s_tir/backend.rst
+42
View File
@@ -0,0 +1,42 @@
.. 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.
tvm.s_tir.backend
-----------------
tvm.s_tir.backend
*****************
.. automodule:: tvm.s_tir.backend
:members:
:imported-members:
tvm.s_tir.backend.adreno
************************
.. automodule:: tvm.s_tir.backend.adreno
:members:
:imported-members:
tvm.s_tir.backend.adreno.pipeline
*********************************
.. automodule:: tvm.s_tir.backend.adreno.pipeline
:members:
tvm.s_tir.backend.adreno.transform
***********************************
.. automodule:: tvm.s_tir.backend.adreno.transform
:members:
:imported-members:
docs/reference/api/python/s_tir/dlight.rst
+66
View File
@@ -0,0 +1,66 @@
.. 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.
tvm.s_tir.dlight
----------------
tvm.s_tir.dlight
****************
.. automodule:: tvm.s_tir.dlight
:members:
:imported-members:
tvm.s_tir.dlight.gpu
********************
.. automodule:: tvm.s_tir.dlight.gpu
:members:
:imported-members:
:noindex:
tvm.s_tir.dlight.adreno
***********************
.. automodule:: tvm.s_tir.dlight.adreno
:members:
:imported-members:
:noindex:
tvm.s_tir.dlight.cpu
********************
.. automodule:: tvm.s_tir.dlight.cpu
:members:
:imported-members:
:noindex:
tvm.s_tir.dlight.analysis
*************************
.. automodule:: tvm.s_tir.dlight.analysis
:members:
:imported-members:
:noindex:
tvm.s_tir.dlight.base
*********************
.. automodule:: tvm.s_tir.dlight.base
:members:
:noindex:
tvm.s_tir.dlight.benchmark
**************************
.. automodule:: tvm.s_tir.dlight.benchmark
:members:
:imported-members:
:noindex:
docs/reference/api/python/s_tir/schedule.rst
+22
View File
@@ -0,0 +1,22 @@
.. 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.
tvm.s_tir.schedule
------------------
.. automodule:: tvm.s_tir.schedule
:members:
:imported-members:
docs/reference/api/python/s_tir/tensor_intrin.rst
+73
View File
@@ -0,0 +1,73 @@
.. 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.
tvm.s_tir.tensor_intrin
-----------------------
tvm.s_tir.tensor_intrin
***********************
.. automodule:: tvm.s_tir.tensor_intrin
:members:
:imported-members:
tvm.s_tir.tensor_intrin.cuda
****************************
.. automodule:: tvm.s_tir.tensor_intrin.cuda
:members:
:noindex:
tvm.s_tir.tensor_intrin.arm_cpu
*******************************
.. automodule:: tvm.s_tir.tensor_intrin.arm_cpu
:members:
:noindex:
tvm.s_tir.tensor_intrin.x86
****************************
.. automodule:: tvm.s_tir.tensor_intrin.x86
:members:
:noindex:
tvm.s_tir.tensor_intrin.rocm
*****************************
.. automodule:: tvm.s_tir.tensor_intrin.rocm
:members:
:noindex:
tvm.s_tir.tensor_intrin.metal
*****************************
.. automodule:: tvm.s_tir.tensor_intrin.metal
:members:
:noindex:
tvm.s_tir.tensor_intrin.hexagon
*******************************
.. automodule:: tvm.s_tir.tensor_intrin.hexagon
:members:
:noindex:
tvm.s_tir.tensor_intrin.riscv_cpu
*********************************
.. automodule:: tvm.s_tir.tensor_intrin.riscv_cpu
:members:
:noindex:
tvm.s_tir.tensor_intrin.dot_product_common
******************************************
.. automodule:: tvm.s_tir.tensor_intrin.dot_product_common
:members:
:noindex:
docs/reference/api/python/s_tir/transform.rst
+23
View File
@@ -0,0 +1,23 @@
.. 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.
tvm.s_tir.transform
-------------------
.. automodule:: tvm.s_tir.transform
:members:
:imported-members:
:no-index:
docs/reference/api/python/script/ir_builder.rst
+46
View File
@@ -0,0 +1,46 @@
.. 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.
tvm.script.ir_builder
---------------------
tvm.script.ir_builder
*********************
.. automodule:: tvm.script.ir_builder
:members:
:imported-members:
tvm.script.ir_builder.ir
************************
.. automodule:: tvm.script.ir_builder.ir
:members:
:imported-members:
tvm.script.ir_builder.relax
***************************
.. automodule:: tvm.script.ir_builder.relax
:members:
tvm.script.ir_builder.relax.distributed
***************************************
.. automodule:: tvm.script.ir_builder.relax.distributed
:members:
tvm.script.ir_builder.tirx
**************************
.. automodule:: tvm.script.ir_builder.tirx
:members:
docs/reference/api/python/script/parser.rst
+37
View File
@@ -0,0 +1,37 @@
.. 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.
tvm.script.parser
-----------------
tvm.script.parser
*****************
.. automodule:: tvm.script.parser
:members:
:imported-members:
tvm.script.parser.ir
********************
.. automodule:: tvm.script.parser.ir
tvm.script.parser.relax
***********************
.. automodule:: tvm.script.parser.relax
tvm.script.parser.tirx
**********************
.. automodule:: tvm.script.parser.tirx
docs/reference/api/python/script/printer.rst
+25
View File
@@ -0,0 +1,25 @@
.. 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.
tvm.script.printer
------------------
tvm.script.printer
******************
.. automodule:: tvm.script.printer
:members:
:imported-members:
docs/reference/api/python/script/script.rst
+35
View File
@@ -0,0 +1,35 @@
.. 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.
tvm.script
----------
tvm.script
**********
.. automodule:: tvm.script
:members:
:imported-members:
tvm.script.relax
****************
.. automodule:: tvm.script.relax
:members:
tvm.script.tirx
***************
.. automodule:: tvm.script.tirx
:members:
docs/reference/api/python/support.rst
+73
View File
@@ -0,0 +1,73 @@
.. 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.
tvm.support
-----------
.. automodule:: tvm.support
:members:
:imported-members:
:autosummary:
tvm.support.cc
~~~~~~~~~~~~~~
.. automodule:: tvm.support.cc
:members:
tvm.support.nvcc
~~~~~~~~~~~~~~~~
.. automodule:: tvm.support.nvcc
:members:
tvm.support.rocm
~~~~~~~~~~~~~~~~
.. automodule:: tvm.support.rocm
:members:
tvm.support.ndk
~~~~~~~~~~~~~~~
.. automodule:: tvm.support.ndk
:members:
tvm.support.xcode
~~~~~~~~~~~~~~~~~
.. automodule:: tvm.support.xcode
:members:
tvm.support.clang
~~~~~~~~~~~~~~~~~
.. automodule:: tvm.support.clang
:members:
tvm.support.emcc
~~~~~~~~~~~~~~~~
.. automodule:: tvm.support.emcc
:members:
tvm.support.popen_pool
~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: tvm.support.popen_pool
:members:
tvm.support.utils
~~~~~~~~~~~~~~~~~
.. automodule:: tvm.support.utils
:members:
tvm.support.tar
~~~~~~~~~~~~~~~
.. automodule:: tvm.support.tar
:members:
docs/reference/api/python/target.rst
+29
View File
@@ -0,0 +1,29 @@
.. 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.
tvm.target
----------
.. automodule:: tvm.target
:members:
:imported-members:
:autosummary:
tvm.target.tag
~~~~~~~~~~~~~~
.. automodule:: tvm.target.tag
:members:
:imported-members:
docs/reference/api/python/te.rst
+25
View File
@@ -0,0 +1,25 @@
.. 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.
tvm.te
------
.. Exclude the ops imported from tirx.
.. automodule:: tvm.te
:members:
:imported-members:
:autosummary:
docs/reference/api/python/testing.rst
+22
View File
@@ -0,0 +1,22 @@
.. 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.
tvm.testing
-----------
.. automodule:: tvm.testing
:members:
:imported-members:
docs/reference/api/python/topi.rst
+65
View File
@@ -0,0 +1,65 @@
.. 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.
tvm.topi
--------
.. automodule:: tvm.topi
:members:
:imported-members:
:noindex:
:autosummary:
tvm.topi.nn
~~~~~~~~~~~
.. automodule:: tvm.topi.nn
:members:
:imported-members:
:autosummary:
tvm.topi.image
~~~~~~~~~~~~~~
.. automodule:: tvm.topi.image
:members:
:imported-members:
:autosummary:
tvm.topi.vision
~~~~~~~~~~~~~~~
.. automodule:: tvm.topi.vision
:members:
:imported-members:
:noindex:
tvm.topi.gpu
~~~~~~~~~~~~
.. automodule:: tvm.topi.gpu
:members:
:imported-members:
:noindex:
tvm.topi.cpp
~~~~~~~~~~~~
.. automodule:: tvm.topi.cpp
:members:
:imported-members:
tvm.topi.testing
~~~~~~~~~~~~~~~~
.. automodule:: tvm.topi.testing
:members:
:imported-members:
docs/reference/api/python/transform.rst
+22
View File
@@ -0,0 +1,22 @@
.. 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.
tvm.ir.transform
-----------------
.. automodule:: tvm.ir.transform
:members:
:imported-members:
docs/reference/publications.rst
+84
View File
@@ -0,0 +1,84 @@
.. 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.
Publications
============
TVM is developed as part of peer-reviewed research in machine learning compiler
framework for CPUs, GPUs, and machine learning accelerators.
This document includes references to publications describing the research,
results, and design that use or built on top of TVM.
2018
* `TVM: An Automated End-to-End Optimizing Compiler for Deep Learning`__, [Slides_]
.. __: https://arxiv.org/abs/1802.04799
.. _Slides: https://www.usenix.org/system/files/osdi18-chen.pdf
* `Learning to Optimize Tensor Programs`__, [Slides]
.. __: https://arxiv.org/pdf/1805.08166.pdf
2020
* `Ansor: Generating High-Performance Tensor Programs for Deep Learning`__, [Slides__] [Tutorial__]
.. __: https://arxiv.org/abs/2006.06762
.. __: https://www.usenix.org/sites/default/files/conference/protected-files/osdi20_slides_zheng.pdf
.. __: https://tvm.apache.org/2021/03/03/intro-auto-scheduler
2021
* `Nimble: Efficiently Compiling Dynamic Neural Networks for Model Inference`__, [Slides__]
.. __: https://arxiv.org/abs/2006.03031
.. __: https://shenhaichen.com/slides/nimble_mlsys.pdf
* `Cortex: A Compiler for Recursive Deep Learning Models`__, [Slides__]
.. __: https://arxiv.org/pdf/2011.01383.pdf
.. __: https://mlsys.org/media/mlsys-2021/Slides/1507.pdf
* `UNIT: Unifying Tensorized Instruction Compilation`__, [Slides]
.. __: https://arxiv.org/abs/2101.08458
* `Lorien: Efficient Deep Learning Workloads Delivery`__, [Slides]
.. __: https://assets.amazon.science/c2/46/2481c9064a8bbaebcf389dd5ad75/lorien-efficient-deep-learning-workloads-delivery.pdf
* `Bring Your Own Codegen to Deep Learning Compiler`__, [Slides] [Tutorial__]
.. __: https://arxiv.org/abs/2105.03215
.. __: https://tvm.apache.org/2020/07/15/how-to-bring-your-own-codegen-to-tvm
2022
* `DietCode: Automatic optimization for dynamic tensor program`__, [Slides]
.. __: https://proceedings.mlsys.org/paper/2022/file/fa7cdfad1a5aaf8370ebeda47a1ff1c3-Paper.pdf
* `Bolt: Bridging the Gap between Auto-tuners and Hardware-native Performance`__, [Slides]
.. __: https://proceedings.mlsys.org/paper/2022/file/38b3eff8baf56627478ec76a704e9b52-Paper.pdf
* `The CoRa Tensor Compiler: Compilation for Ragged Tensors with Minimal Padding`__, [Slides]
.. __: https://arxiv.org/abs/2110.10221
docs/reference/security.rst
+83
View File
@@ -0,0 +1,83 @@
.. 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.
.. _dev-security:
Security Guide
==============
Reporting Security Issues
-------------------------
The Apache Software Foundation takes a very active stance in eliminating security problems and denial of service attacks against its products.
We strongly encourage folks to report such problems to our private security mailing list first, before disclosing them in a public forum.
Please note that the security mailing list should only be used for reporting undisclosed security vulnerabilities and managing the process of fixing such vulnerabilities. We cannot accept regular bug reports or other queries at this address. All mail sent to this address that does not relate to an undisclosed security problem in our source code will be ignored.
Questions about: if a vulnerability applies to your particular application obtaining further information on a published vulnerability availability of patches
and/or new releases should be addressed to the user Discuss forum.
The private security mailing address is: `security@apache.org <security@apache.org>`_.
Feel free to consult the `Apache Security guide <https://www.apache.org/security/>`_.
Security Model
--------------
The default binary generated by TVM only relies on a minimum runtime API.
The runtime depends on a limited set of system calls(e.g. malloc) in the system library.
TVM RPC server assumes that the user is trusted and needs to be used in a trusted network environment
and encrypted channels. It allows writings of arbitrary files into the server and provide
full remote code execution capabilities to anyone who can access this API.
All of the TVM APIs are designed to be used by trusted users, for APIs that involves URL,
we expect users to put in only trusted URLs.
RPC data exchange between the tracker, server and client are in plain-text.
It is recommended to use them under trusted networking environment or encrypted channels.
Loading serialized artifacts must only be done with input from a trusted source.
Security Advisories
-------------------
Subroutine Cache Hash Collision
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
``SubroutineMixin._get_subroutine()`` in ``python/tvm/relax/frontend/nn/subroutine.py``
used ``tvm_ffi.structural_hash`` as the sole cache lookup key without a subsequent
``structural_equal`` verification. If two different ``arg_ty`` values produced the
same 64-bit hash, the cache would return a previously compiled function with
mismatched parameter shapes, leading to silently incorrect compiled output.
**Severity**: Low. The ``structural_hash`` function returns a 64-bit integer.
A natural hash collision requires approximately 2^32 distinct inputs (birthday bound),
making accidental collision extremely unlikely in normal compilation workflows.
The issue is primarily a correctness defect rather than a practically exploitable
security vulnerability.
**Root Cause**: The subroutine cache (``cls._gvar``) was keyed by
``(structural_hash(arg_ty, map_free_vars=True), is_dataflow)``.
A hash match was treated as proof of structural equality, skipping the necessary
``structural_equal`` check.
**Fix**: The cache now stores a list of ``(arg_ty, result)`` pairs per hash bucket.
On lookup, each candidate is verified with ``structural_equal`` before returning.
This follows the standard hash-table pattern: hash for bucket selection, equality
for final verification.