手写TVM算子
@tvm.script.ir_module
class MyConv:
@T.prim_func
def conv(A: T.Buffer((N, CI, H, W), "int64"),
B: T.Buffer((CO, CI, K, K), "int64"),
C: T.Buffer((N, CO, H-K+1, W-K+1), "int64")):
T.func_attr({"global_symbol": "conv", "tir.noalias": True})
for n, c, h, w...