Skip to content

Commit

Permalink
[DOC] Developer documentation for InferBound pass. (apache#3126)
Browse files Browse the repository at this point in the history
* Developer documentation for InferBound pass.
  • Loading branch information
jdavies-huawei authored and Wei Chen committed May 13, 2019
1 parent 8e07746 commit 5775d1a
Show file tree
Hide file tree
Showing 3 changed files with 777 additions and 1 deletion.
5 changes: 4 additions & 1 deletion docs/dev/codebase_walkthrough.rst
Original file line number Diff line number Diff line change
Expand Up @@ -136,7 +136,10 @@ Lowering is done by ``tvm.lower()`` function, defined in ``python/tvm/build_modu
stmt = schedule.ScheduleOps(sch, bounds)
...

Bound inference is the process where all loop bounds and sizes of intermediate buffers are inferred. If you target the CUDA backend and you use shared memory, its required minimum size is automatically determined here. Bound inference is implemented in ``src/schedule/bound.cc``, ``src/schedule/graph.cc`` and ``src/schedule/message_passing.cc``.
Bound inference is the process where all loop bounds and sizes of intermediate buffers are inferred. If you target the CUDA backend and you use shared memory, its required minimum size is automatically determined here. Bound inference is implemented in ``src/schedule/bound.cc``, ``src/schedule/graph.cc`` and ``src/schedule/message_passing.cc``. For more information on how bound inference works, see `InferBound Pass`_.

.. _InferBound Pass: http://docs.tvm.ai/dev/inferbound.html


``stmt``, which is the output of ``ScheduleOps()``, represents an initial loop nest structure. If you have applied ``reorder`` or ``split`` primitives to your schedule, then the initial loop nest already reflects that changes. ``ScheduleOps()`` is defined in ``src/schedule/schedule_ops.cc``.

Expand Down
1 change: 1 addition & 0 deletions docs/dev/index.rst
Original file line number Diff line number Diff line change
Expand Up @@ -33,3 +33,4 @@ In this part of documentation, we share the rationale for the specific choices m
relay_add_op
relay_add_pass
codebase_walkthrough
inferbound
Loading

0 comments on commit 5775d1a

Please sign in to comment.