Enable GTP for heterogeneous MIMO training (#6570)