add module setup function in pipe stage