use custom partitioning for coarse scale op too