add support for all 3 graphs for collective