def METHOD_NAME(self):
    class Model(torch.nn.Module):
        def __init__(self):
            super().__init__()
            self.w = torch.nn.Parameter(
                torch.ones(2, 3, device=torch.device("cpu"))
            )
            self.b = torch.nn.Parameter(torch.ones(3, device=torch.device("cuda")))
        def forward(self, x, y):
            return torch.matmul(self.w, x), y + self.b
    x = torch.randn(3, 3, device=torch.device("cpu"))
    y = torch.randn(3, 3, device=torch.device("cuda"))
    self.run_test(Model(), (x, y))