something is different from given notebook

[mazdarx7fc3s]

func = func.with_attr("global_symbol", "main")
ir_mod_from_te = IRModule({"main": func})
print(ir_mod_from_te.script())

My result:

@tvm.script.ir_module
class Module:
    @tir.prim_func
    def func(A: tir.Buffer[128, "float32"], B: tir.Buffer[128, "float32"], C: tir.Buffer[128, "float32"]) -> None:
        # function attr dict
        tir.func_attr({"global_symbol": "main", "tir.noalias": True})
        # body
        # with tir.block("root")
        for i0 in tir.serial(128):
            with tir.block("C"):
                i = tir.axis.spatial(128, i0)
                tir.reads(A[i], B[i])
                tir.writes(C[i])
                C[i] = A[i] + B[i]

The func is named as 'func', is not 'main'.

[Hzfengsy]

Thanks for pointing it out. It is because a small bug fix was merged after the notebook publishing. Unfortunately, it brings this bug of the incorrect function name. We will try to fix it ASAP.