Beya2019 commented on a change in pull request #8408:
URL: https://github.com/apache/tvm/pull/8408#discussion_r666030868
##########
File path: tests/python/unittest/test_tvmscript_complete.py
##########
@@ -190,9 +190,77 @@ def render(e):
assert False
[email protected]
+def func_with_bufferslice_indices(data: ty.handle, index: ty.handle) -> None:
+ data_buf = tir.match_buffer(data, (16, 16), "float32")
+ index_buf = tir.match_buffer(index, (1,), "int32")
+ out_buf = tir.alloc_buffer((16, 16), "float32")
+
+ with tir.block([16, 16]) as [vi, vj]:
+ out_buf[vi, vj] = data_buf[vi, index_buf[0]]
+
+
[email protected]
+def expected_bufferslice_indices(data: ty.handle, index: ty.handle) -> None:
+ index_buf = tir.match_buffer(
Review comment:
OK, Get it. Thanks for reminding.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
