KireinaHoro opened a new pull request #5050: [CODEGEN] Add support for extern prologue/epilogue functions URL: https://github.com/apache/incubator-tvm/pull/5050 This commmit adds support for adding prologue and/or epilogue function calls for a code block via the pragma schedule. A dummy use can be found here: https://gist.github.com/KireinaHoro/c959170557d75d81c52c116c3750d78a The addition is proposed to support better control in the generated code when using the tensorize and inline asm flows to do codegen for something like [Gemmini](https://github.com/ucb-bar/gemmini), which requires explicit `fence` instructions to block the execution flow until data has been fully flushed back to DRAM. With this change an external function that does the `fence` can be placed in the epilogue of a computation. The prologue pragma may be useful in similar situations when a device needs some sort of initialization, and is added for symmetry.
---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected] With regards, Apache Git Services
