[PD] GEM support in GridFlow