Thanks @xiacijie for the question!
```python
# Snippet-1:
for i, j in tir.grid(128, 128):
with tir.block([128, 128], "init") as [vi, vj]:
C[vi, vj] = 0.0
# Snippet-2:
with tir.block([128, 128], "init") as [vi, vj]:
C[vi, vj] = 0.0
```
Yes, the two snippets above are strictly equivalent, and the second one is the
syntactic sugar for the first. TVM script has a built-in functionality called
"auto-completion" that desugars snippets.
To get a sense what the fully desugared IR looks like, you may print it out
with the following command in python:
```python
print(tvm.script.asscript(PrimFunc-or-IRModule))
```
---
[Visit
Topic](https://discuss.tvm.apache.org/t/combining-separate-tir-block-using-compute-at/10113/10)
to respond.
You are receiving this because you enabled mailing list mode.
To unsubscribe from these emails, [click
here](https://discuss.tvm.apache.org/email/unsubscribe/928d22fbbccc01d91a9bcfefaf05771787af05ebe7b44e5e2ab89c8e5573fefd).
