Hi @xiaolong18,
Thanks for playing with Relax! print(R.parser.astext(relax_mod))
actually prints out the whole IRModule, which contains both the Relax functions and TIR PrimFuncs. Please search for “@relax.function” in the printed text, you will find the Relax IR as below:
If you just want to see the relax function, you can do print(R.parser.astext(relax_mod["main"]))
.