Support different subsets for each stage