[TVMC] Add option to dump TIR code to file

[lhutton1]

Currently dumps the TIR code after the final phase of lowering (phase 3 from

tvm/gallery/how_to/extend_tvm/low_level_custom_pass.py

Line 152 in 665dd41

# Here are the essential transformations done by each phase:

) before codegen. This is done by running a pass to capture the TIR module as it is not saved during the build. The result is saved to file similar to the other code dumps.

In the future it seems good to have more granularity in the IR that is being dumped, e.g. print IR before/after named pass. I thought this might be helpful in the meantime.

cc @ekalda @ashutosh-arm @neildhickey

[tvm-bot]

Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment.

• cc @Mousius, @gromero, @leandron _{See #10317 for details}

_{Generated by tvm-bot}

[ashutosh-arm]

LGTM! Thank you 😄

[neildhickey]

LGTM too. Thanks

[echuraev]

LGTM. Thanks

[ashutosh-arm]

Thanks @lhutton1, @neildhickey and @echuraev for the commits and the reviews 😄

[PhilippvK]

In the future it seems good to have more granularity in the IR that is being dumped, e.g. print IR before/after named pass. I thought this might be helpful in the meantime.

@lhutton1 I really like these changes. Are you going to do a followup for this soon? If not it would be great if we could open up an issue to collect some ideas. I plan to do something similar for the RelayIR dump (allow dumping post/pre-partitioning IR) and would like to follow a similar approach.

[lhutton1]

Hi @PhilippvK, thanks for the interest! I'm hoping to have a follow-up soon (~1wk) but I have another commitment I need to work on first. I'll summarise the thoughts I have in a discuss post - it would be good to see if they align to your thoughts/requirements before diving in