wrongtest commented on PR #11097:
URL: https://github.com/apache/tvm/pull/11097#issuecomment-1107956829
If we'd like to distinguish variables and other asts in script and variables
in host language? For example, use a mustache syntax the case above could be
(like the html template of VUE.js)
```python
def matmul_generator(M, N, K, dtype):
@T.prim_func
def matmul(a: T.handle, b: T.handle, c: T.handle) -> None:
A = T.match_buffer(a, [{M}, {K}], dtype={dtype})
B = T.match_buffer(b, [{N}, {K}], dtype={dtype})
C = T.match_buffer(c, [{M}, {N}], dtype={dtype})
for i, j, k in T.grid({M}, {N}, {K}):
with T.block():
vi, vj, vk = T.axis.remap("SSR", [i, j, k])
with T.init():
C[vi, vj] = T.float32(0)
C[vi, vj] = C[vi, vj] + A[vi, vk] * B[vj, vk]
return matmul
```
And we can also write like
```python
T.evaluate({ [call my expr ib builder] })
```
or
```python
for i in range({N}):
{
[call my ib builder to create sequence of stmts]
}
```
--
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]
