- if (TRACE_smpi_is_enabled() && TRACE_smpi_is_grouped())
- container->type_->getOrCreateContainerType("MPI")->getOrCreateStateType("MPI_STATE");
+ if (TRACE_smpi_is_enabled() && TRACE_smpi_is_grouped()) {
+ simgrid::instr::ContainerType* mpi = container->type_->getOrCreateContainerType("MPI");
+ mpi->getOrCreateStateType("MPI_STATE");
+ // TODO See if we can move this to the LoadBalancer plugin
+ root->type_->getOrCreateLinkType("MIGRATE_LINK", mpi, mpi);
+ mpi->getOrCreateStateType("MIGRATE_STATE");
+ }