Skip to content

Commit

Permalink
Fix sphinx warning
Browse files Browse the repository at this point in the history
  • Loading branch information
MasterJH5574 committed Jul 29, 2021
1 parent ec86ea5 commit b20381b
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions python/tvm/tir/schedule/schedule.py
Original file line number Diff line number Diff line change
Expand Up @@ -541,7 +541,6 @@ def rfactor(self, loop: LoopRV, factor_axis: int) -> LoopRV:
- The intermediate buffer, or "rf-buffer" is a buffer of rank `ndim(B) + 1` and
size `size(B) * n`, whose shape expands from `shape(B)` by adding an axis of `n`
at the position specified by `factor_axis`. For example,
* shape(B) = [1, 2, 3], factor_axis = 0 => shape(B_rf) = [n, 1, 2, 3]
* shape(B) = [1, 2, 3], factor_axis = 1 => shape(B_rf) = [1, n, 2, 3]
* shape(B) = [1, 2, 3], factor_axis = 2 => shape(B_rf) = [1, 2, n, 3]
Expand All @@ -550,7 +549,6 @@ def rfactor(self, loop: LoopRV, factor_axis: int) -> LoopRV:
- The rfactor block, or "rf-block", is a block that writes to the `rf-buffer` without
accumulating over the loop `k`, i.e. the loop `k` is converted from a reduction loop
to a data parallel loop. In our example, the rf-block is:
.. code-block:: python
B_rf = np.zeros((128, 128)) # the rf-buffer
Expand All @@ -562,7 +560,6 @@ def rfactor(self, loop: LoopRV, factor_axis: int) -> LoopRV:
- The write-back block, or `wb-block`, is a block that accumulates the rf-buffer into
the result buffer. All the reduction loops are removed except the loop `k` for accumulation.
In our example, the wb-block is:
.. code-block:: python
for i in range(128): # loop i is a data parallel loop (unchanged)
Expand Down

0 comments on commit b20381b

Please sign in to comment.