[auto_group] Adding reduction deps to pk_map