Support kernel splitting in code generation