Yeah I mean honestly I think this is the proper solution long-term. The question is whether you should have to import tvm to parse tir, which is true now (e.g. via TVMScript). Also TVMScript does rely on Python AST parsing and that’s another thing which may not be desirable if we make TIR an exported language. At this point I was just trying to unblock the save_json approach in case TVMScript isn’t usable here, and ran into this bug.
1 Like