Extend FMM performance model to allow load balance data gathering