Hello, assuming I performed my scheduling in TIR using the functions related to tir.Schedule, can I somehow use the result as the main function in an IRModule and use that to then call relay.build? Or is this not currently supported? Thanks in advance!
I have tried to do this:
tvm.relay.build(mod, tvm.target.target.stm32("stm32F7xx"))
But that results in an error:
Check failed: (can_dispatch(n)) is false: NodeFunctor calls un-registered function on type tir.PrimFunc
I guess one possible solution is using relay BYOC to inject the code that were generated by your scheduled tir, otherwise maybe you can take a look at relax.
Does Relax support baremetal environments? I don’t think that is the case, which is why I decided to go with Relay. I’d love to be wrong here though and move over to the newer infrastructure.