How do I dump the IRs using PassContext?

kausthubtm · October 5, 2022, 8:14am

How to dump IRs for each pass when building a model - #8 by leeexyz

This discussion gives details about dumping IRs using the tvm.build() function but this has been depreciated. Is there a work around for the same using PassContext or without installing the older version from source and using tvm.build() ?

sunggg · October 7, 2022, 6:23pm

Hi, @kausthubtm Is this close to what you are looking for?

github.com

tlc-pack/relax/blob/relax/tests/python/relay/test_pass_instrument.py#L66


    mod = tvm.relay.transform.InferType()(mod)


    profiles = pass_timing.render()
    assert profiles == ""




instrument_definition_type = tvm.testing.parameter("decorator", "subclass")




def test_custom_instrument(instrument_definition_type):
    class BaseTest:
        def __init__(self):
            self.events = []


        def enter_pass_ctx(self):
            self.events.append("enter ctx")


        def exit_pass_ctx(self):
            self.events.append("exit ctx")


        def run_before_pass(self, mod, info):

By using instrumentation, you can print mod before and after each pass.