* [BUILD/CODEGEN] Allow combine multiple functions in build stage.

* Enhance code module

* fix compile