fit pipeline for shardformer policy as discussed in #4050
fit pipeline for shardformer policy as discussed in #4050