Add a flag for single code gen module per code gen task