Ooh, good catch, sorry for this. I mostly left that check in as a debug tool for developing passes, but you’re right, the partitioned graph won’t match anymore if we’re doing more specific things, like matching on names, or possibly matching on types. I think maybe we should just remove the check, or leave it there but comment it out so it’s only used for pass development, not user-facing workloads.