(x: R.Tensor([1024], "float32"), y: R.Tensor([1024], "float32"))
| 35 | class Mod: |
| 36 | @R.function |
| 37 | def main(x: R.Tensor([1024], "float32"), y: R.Tensor([1024], "float32")): |
| 38 | lv0 = R.add(x, y) |
| 39 | return lv0 |
| 40 | |
| 41 | mod = pipeline(Mod) |
| 42 | sch = tvm.s_tir.Schedule(mod) |