val shuffleRead = listener.stageToShuffleRead.exists {case (k, v) => k == stageId && v > 0)
val shuffleWrite = listener.stageToShuffleWrite.exists {case (k, v) => k == stageId && v > 0)