I used unity branch code to build a tvm with relax, but I ran the following code sample incorrectly:
from __future__ import annotations
import numpy as np
import tvm
from tvm import relax, topi
from tvm.ir.module import IRModule
from tvm.script import relax as R
from tvm.script import tir as T
# from torch import Tensor
@tvm.script.ir_module
class MyModule:
@R.function
def main(x: Tensor((3, 4), "float32"), y: Tensor((3, 4), "float32")):
with relax.dataflow():
lv0 = relax.multiply(x, y)
gv0 = relax.add(lv0, y)
relax.output(gv0)
return gv0
relax_func = MyModule["main"]
print(relax_func.params)
下面是我的报错截图