[BUG] Dataset shuffle is not perfect

[goodwanghan]

Describe the bug
It only shuffles within partition, we need a complete shuffle, but the implementation is not working in dask, we need to fix the issue of per_row that does not work well for dask dataframe.