_cumsum moves the data to default GPU when deterministic flag is on

[KubaMichalczyk]

If you're working with multiple GPUs and the input tensor x is not on cuda:0, the following line causes the tensor to land on the default GPU. This causes a mismatch in any later call on x, as other tensors are likely still on the original GPU.

torchmetrics/src/torchmetrics/utilities/data.py

Line 218 in 50e7da3

return x.cpu().cumsum(dim=dim, dtype=dtype).cuda()

A simple reference to x.device can solve this problem:

return x.cpu().cumsum(dim=dim, dtype=dtype).cuda(x.device)

[github-actions]

Hi! thanks for your contribution!, great first issue!