(x: T.Buffer((10,), "float32"), y: T.Buffer((10,), "float32"))
| 2453 | class Before: |
| 2454 | @T.prim_func(private=True, s_tir=True) |
| 2455 | def add1(x: T.Buffer((10,), "float32"), y: T.Buffer((10,), "float32")): |
| 2456 | T.func_attr({"tirx.noalias": True}) |
| 2457 | for i in range(10): |
| 2458 | with T.sblock("compute1"): |
| 2459 | vi = T.axis.spatial(10, i) |
| 2460 | y[vi] = x[vi] + T.float32(1.0) |
| 2461 | |
| 2462 | @T.prim_func(private=True, s_tir=True) |
| 2463 | def mul1(x: T.Buffer((10,), "float32"), y: T.Buffer((10,), "float32")): |