stateless.functional_call doesn't work with nn.DataParallel

[Chillee]

🐛 Describe the bug

Encountered in #77137 (comment)

import torch
import torch.nn as nn
import torch.nn.utils.stateless as stateless

class Foo(nn.Module):
    def __init__(self):
        super().__init__()
        self.weight = nn.Parameter(torch.ones(5))

    def forward(self, x):
        return self.weight + x

mod = Foo().cuda()
mod = nn.DataParallel(mod, [0, 1])
print(stateless.functional_call(mod, {'module.weight': torch.zeros(5, device='cuda')}, (torch.ones(2, 5, device='cuda'),)))

errors with
``
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cuda:1!

### Versions

Nightly

cc @ezyang @gchanan @zou3519 @albanD @mruberry @jbschlosser @walterddr @kshitij12345

[albanD]

I think this should at least be a loud error as it is unlikely we're going to fix it right now.

[jbschlosser]

Actionable to add a loud error.

[kurtamohler]

I've added the error. Should we remove the high priority label, or is fixing it also high priority?