[XLA:GPU] Match more sort cases in the GPU Sort Rewriter.
[XLA:GPU] Match more sort cases in the GPU Sort Rewriter.

The additional matching logic specifically covers the common output of the StableSortExpander pass. This change also enhances the `GpuSortRewriter` tests to actually run the rewrites and ensure they are correct.
