Correlation is zero in Multi-GPU environment

[KangBK0120]

Hi, first of all, thanks for sharing this module.

the correlation value is always zero in the muti-GPU environment when the inputs are not in the first GPU.

totally works fine as expected.

However, if I try to use other GPUs, it returns 0 correlation.

And by using CUDA_VISIBLE_DEVICES" to point a single GPU, this problem is gone.

[KangBK0120]

I'm using torch 1.9.0 and 3090 RTX with cuda version 11.1

[ClementPinard]

Hi, did you try to install it from source ? There was a discussion about this problem a while back and was solved by @InnovArul #59