This is an automated email from the ASF dual-hosted git repository.
junrushao pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new 9e5055b358 [TIR] [Docs] Fix unsafe_set_dtype docstring (#14611)
9e5055b358 is described below
commit 9e5055b358250542742b429c35cc0dc98e7064f9
Author: Anirudh Sundar Subramaniam <[email protected]>
AuthorDate: Thu Apr 13 21:13:06 2023 +0530
[TIR] [Docs] Fix unsafe_set_dtype docstring (#14611)
Looks like the `unsafe_set_dtype` docs were not updated after the API was
renamed during PR reviews, just fixing that.
---
python/tvm/tir/schedule/schedule.py | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)
diff --git a/python/tvm/tir/schedule/schedule.py
b/python/tvm/tir/schedule/schedule.py
index 34fd649a5d..11fa7ae12e 100644
--- a/python/tvm/tir/schedule/schedule.py
+++ b/python/tvm/tir/schedule/schedule.py
@@ -2581,7 +2581,7 @@ class Schedule(Object):
Examples
--------
- Before set_dtype, in TensorIR, the IR is:
+ Before unsafe_set_dtype, in TensorIR, the IR is:
.. code-block:: python
@@ -2600,12 +2600,12 @@ class Schedule(Object):
vi, vj = T.axis.remap("SS", [i, j]
C[vi, vj] = B[vi, vj] + 1.0
- Create the schedule and do set_dtype:
+ Create the schedule and do unsafe_set_dtype:
.. code-block:: python
sch = tir.Schedule(before_set_dtype)
- sch.set_dtype("B", buffer_index=0, dtype="float16")
+ sch.unsafe_set_dtype("B", buffer_index=0, dtype="float16")
print(sch.mod["main"].script())
After applying set_dtype, the IR becomes:
@@ -2629,7 +2629,8 @@ class Schedule(Object):
Note
----
- `set_dtype` requires the buffer to be an intermediate buffer defined
via `alloc_buffer`.
+ `unsafe_set_dtype` requires the buffer to be an intermediate buffer
defined via
+ `alloc_buffer`.
"""
block = self._normalize_block_arg(block)
_ffi_api.ScheduleUnsafeSetDType( # type: ignore # pylint:
disable=no-member
