Torch.prod produces RuntimeError: CUDA driver error: invalid argument

Issue solved after remove files under ~/.cache/torch/kernels/.

It’s weird, anybody could explain this code-wise ?