Hi all.
I have used auto_scheduler to tune a network on CPU. I can run the network with relay.build. My question is how to dump the assembly code for each operator? There might be two steps: identify each operator in the graph and then dump source code for each operator. Thanks!
After you do lib = relay.build(...), you can do lib.lib.get_source("asm") to dump the asm.
Thanks for quick reply. Do you know if I can dump asm code for each operator instead of getting all asm code at a time?
Since we compile all operators at once, we cannot dump operator individually. But you can find the asm of your interest in the big asm dump by searching for operator name (look for *_compute)