def test_param_on_pointer(workers):
tensor = torch.tensor([1.0, -1.0, 3.0, 4.0])
ptr = tensor.send(workers["bob"])
param = Parameter(ptr)
local_param = param.get()
assert (local_param.data == tensor).all()
def test_param_double_send_get(workers):
tensor = torch.tensor([[1.0, 1]])
param = Parameter(tensor)
param = param.send(workers["bob"]).send(workers["alice"])
param = param.get().get()
assert (param.data == tensor).all()
