As for flaky, I admit a little bit.   I believe the order of calling `materialize` can guarantee that the order of task to be scheduled in normal circumstances, but, to be honest, this is guarantee is not strict since the submit of broadcast job and shuffle map stage(job) are in different thread. But, at least we reach the same level as non AQE.