Update DumpCallGraph to new multi-stage reductions API