nn.Sort should also work with cuda
compiles. Sort test passes