def METHOD_NAME(self):
    """Test goes to GPU for matrix valued inputs."""
    dev = qml.device("default.qubit.torch", wires=1)
    U = torch.eye(2, requires_grad=False, device=torch.device("cuda"))
    with qml.queuing.AnnotatedQueue() as q:
        qml.QubitUnitary(U, wires=0)
        qml.expval(qml.PauliZ(0))
    tape = qml.tape.QuantumScript.from_queue(q)
    res = dev.execute(tape)
    assert res.is_cuda
    assert "cuda" in dev._torch_device