def test_distributed_circuit_addition():
# Attempt to add circuits with different devices
original_backend = qibo.get_backend()
qibo.set_backend("custom")
devices = {"/GPU:0": 2, "/GPU:1": 2}
c1 = models.DistributedCircuit(6, devices)
c2 = models.DistributedCircuit(6, {"/GPU:0": 2})
with pytest.raises(ValueError):
c = c1 + c2
c2 = models.DistributedCircuit(6, devices)
c1.add([gates.H(i) for i in range(6)])
c2.add([gates.CNOT(i, i + 1) for i in range(5)])
c2.add([gates.Z(i) for i in range(6)])
dist_c = c1 + c2
c = models.Circuit(6)
c.add([gates.H(i) for i in range(6)])
c.add([gates.CNOT(i, i + 1) for i in range(5)])
c.add([gates.Z(i) for i in range(6)])
target_state = c().numpy()
final_state = dist_c().numpy()
assert c.depth == dist_c.depth
np.testing.assert_allclose(target_state, final_state)
qibo.set_backend(original_backend)
def test_default_initialization():
devices = {"/GPU:0": 2, "/GPU:1": 2}
c = models.DistributedCircuit(6, devices)
c.queues.qubits = distutils.DistributedQubits(range(c.nglobal), c.nqubits)
state = c.get_initial_state()
final_state = state.numpy()
target_state = np.zeros_like(final_state)
target_state[0] = 1
np.testing.assert_allclose(final_state, target_state)
def test_set_gates_simple():
devices = {"/GPU:0": 2, "/GPU:1": 2}
c = models.DistributedCircuit(6, devices)
c.add((gates.H(i) for i in range(4)))
c.queues.set(c.queue)
check_device_queues(c.queues)
assert len(c.queues.queues) == 1
assert len(c.queues.queues[0]) == 4
for queues in c.queues.queues[0]:
assert len(queues) == 4
def test_user_initialization(nqubits):
import itertools
target_state = utils.random_numpy_state(nqubits)
devices = {"/GPU:0": 2, "/GPU:1": 2}
c = models.DistributedCircuit(nqubits, devices)
c.queues.qubits = distutils.DistributedQubits(range(c.nglobal), c.nqubits)
state = c.get_initial_state(target_state)
np.testing.assert_allclose(state.numpy(), target_state)
target_state = target_state.reshape(nqubits * (2, ))
for i, s in enumerate(itertools.product([0, 1], repeat=c.nglobal)):
piece = state.pieces[i].numpy()
target_piece = target_state[s]
np.testing.assert_allclose(target_piece.ravel(), piece)
def test_simple_execution_global(ndevices):
original_backend = qibo.get_backend()
qibo.set_backend("custom")
devices = {"/GPU:0": ndevices // 2, "/GPU:1": ndevices // 2}
dist_c = models.DistributedCircuit(6, devices)
dist_c.add((gates.H(i) for i in range(6)))
c = models.Circuit(6)
c.add((gates.H(i) for i in range(6)))
initial_state = utils.random_numpy_state(c.nqubits)
final_state = dist_c(np.copy(initial_state)).numpy()
target_state = c(np.copy(initial_state)).numpy()
np.testing.assert_allclose(target_state, final_state)
qibo.set_backend(original_backend)
def test_controlled_execution(ndevices):
original_backend = qibo.get_backend()
qibo.set_backend("custom")
devices = {"/GPU:0": ndevices}
dist_c = models.DistributedCircuit(4, devices)
dist_c.add((gates.H(i) for i in range(dist_c.nglobal, 4)))
dist_c.add(gates.CNOT(0, 2))
c = models.Circuit(4)
c.add((gates.H(i) for i in range(dist_c.nglobal, 4)))
c.add(gates.CNOT(0, 2))
initial_state = utils.random_numpy_state(c.nqubits)
final_state = dist_c(np.copy(initial_state)).numpy()
target_state = c(np.copy(initial_state)).numpy()
np.testing.assert_allclose(target_state, final_state)
qibo.set_backend(original_backend)
def test_set_gates_controlled():
devices = {"/GPU:0": 2, "/GPU:1": 2}
c = models.DistributedCircuit(6, devices)
c.add([gates.H(0), gates.H(2), gates.H(3)])
c.add(gates.CNOT(4, 5))
c.add(gates.Z(1).controlled_by(0))
c.add(gates.SWAP(2, 3))
c.add([gates.X(2), gates.X(3), gates.X(4)])
c.queues.set(c.queue)
check_device_queues(c.queues)
assert len(c.queues.queues) == 7
for i, queue in enumerate(c.queues.queues[:-2]):
assert len(queue) == 4 * (1 - i % 2)
for device_group in c.queues.queues[0]:
assert len(device_group) == 7
for device_group in c.queues.queues[2]:
assert len(device_group) == 1
def test_transform_queue_simple():
original_backend = qibo.get_backend()
qibo.set_backend("custom")
devices = {"/GPU:0": 1, "/GPU:1": 1}
c = models.DistributedCircuit(4, devices)
c.add((gates.H(i) for i in range(4)))
c.queues.qubits = distutils.DistributedQubits([0], c.nqubits)
tqueue = c.queues.transform(c.queue)
assert len(tqueue) == 6
for i in range(3):
assert isinstance(tqueue[i], gates.H)
assert tqueue[i].target_qubits == (i + 1, )
assert isinstance(tqueue[3], gates.SWAP)
assert tqueue[3].target_qubits == (0, 1)
assert isinstance(tqueue[4], gates.H)
assert tqueue[4].target_qubits == (1, )
assert isinstance(tqueue[5], gates.SWAP)
assert tqueue[5].target_qubits == (0, 1)
qibo.set_backend(original_backend)
def test_execution_with_global_swap():
original_backend = qibo.get_backend()
qibo.set_backend("custom")
devices = {"/GPU:0": 2, "/GPU:1": 1, "/GPU:2": 1}
dist_c = models.DistributedCircuit(6, devices)
dist_c.add((gates.H(i) for i in range(6)))
dist_c.add((gates.SWAP(i, i + 1) for i in range(5)))
dist_c.global_qubits = [0, 1]
c = models.Circuit(6)
c.add((gates.H(i) for i in range(6)))
c.add((gates.SWAP(i, i + 1) for i in range(5)))
initial_state = utils.random_numpy_state(c.nqubits)
final_state = dist_c(np.copy(initial_state)).numpy()
target_state = c(np.copy(initial_state)).numpy()
np.testing.assert_allclose(target_state, final_state)
qibo.set_backend(original_backend)
def test_set_gates_with_global_swap():
devices = {"/GPU:0": 2, "/GPU:1": 2}
c = models.DistributedCircuit(6, devices)
c.add([gates.H(0), gates.H(2), gates.H(3)])
c.add(gates.SWAP(3, 4))
c.add([gates.X(1), gates.X(2)])
c.queues.qubits = distutils.DistributedQubits([4, 5], c.nqubits)
c.queues.create(c.queues.transform(c.queue))
check_device_queues(c.queues)
assert len(c.queues.special_queue) == 1
assert len(c.queues.queues) == 3
assert len(c.queues.queues[0]) == 4
assert len(c.queues.queues[1]) == 0
assert len(c.queues.queues[2]) == 4
for device_group in c.queues.queues[0]:
assert len(device_group) == 3
for device_group in c.queues.queues[2]:
assert len(device_group) == 2
def test_controlled_execution_large(ndevices):
devices = {"/GPU:0": ndevices // 2, "/GPU:1": ndevices // 2}
dist_c = models.DistributedCircuit(6, devices)
dist_c.add([gates.H(0), gates.H(2), gates.H(3)])
dist_c.add(gates.CNOT(4, 5))
dist_c.add(gates.Z(1).controlled_by(0))
dist_c.add(gates.SWAP(2, 3))
dist_c.add([gates.X(2), gates.X(3), gates.X(4)])
c = models.Circuit(6)
c.add([gates.H(0), gates.H(2), gates.H(3)])
c.add(gates.CNOT(4, 5))
c.add(gates.Z(1).controlled_by(0))
c.add(gates.SWAP(2, 3))
c.add([gates.X(2), gates.X(3), gates.X(4)])
initial_state = utils.random_numpy_state(c.nqubits)
final_state = dist_c(np.copy(initial_state)).numpy()
target_state = c(np.copy(initial_state)).numpy()
np.testing.assert_allclose(target_state, final_state)
def test_distributed_state_getitem():
theta = np.random.random(4)
c = models.DistributedCircuit(4, {"/GPU:0": 2})
c.add((gates.RX(i, theta=theta[i]) for i in range(4)))
state = c()
c = models.Circuit(4)
c.add((gates.RX(i, theta=theta[i]) for i in range(4)))
target_state = c().numpy()
# Check indexing
state_vector = np.array([state[i] for i in range(2**4)])
np.testing.assert_allclose(state_vector, target_state)
# Check slicing
np.testing.assert_allclose(state[:], target_state)
np.testing.assert_allclose(state[2:5], target_state[2:5])
# Check list indexing
ids = [2, 4, 6]
np.testing.assert_allclose(state[ids], target_state[ids])
# Check error
with pytest.raises(TypeError):
state["a"]
def test_transform_queue_more_gates():
original_backend = qibo.get_backend()
qibo.set_backend("custom")
devices = {"/GPU:0": 2, "/GPU:1": 2}
c = models.DistributedCircuit(4, devices)
c.add(gates.H(0))
c.add(gates.H(1))
c.add(gates.CNOT(2, 3))
c.add(gates.CZ(0, 1))
c.add(gates.CNOT(3, 0))
c.add(gates.CNOT(1, 2))
c.queues.qubits = distutils.DistributedQubits([2, 3], c.nqubits)
tqueue = c.queues.transform(c.queue)
assert len(tqueue) == 10
assert isinstance(tqueue[0], gates.H)
assert tqueue[0].target_qubits == (0, )
assert isinstance(tqueue[1], gates.H)
assert tqueue[1].target_qubits == (1, )
assert isinstance(tqueue[2], gates.CZ)
assert tqueue[2].target_qubits == (1, )
assert isinstance(tqueue[3], gates.SWAP)
assert set(tqueue[3].target_qubits) == {1, 3}
assert isinstance(tqueue[4], gates.CNOT)
assert tqueue[4].target_qubits == (1, )
assert isinstance(tqueue[5], gates.CNOT)
assert tqueue[5].target_qubits == (0, )
assert isinstance(tqueue[6], gates.SWAP)
assert set(tqueue[6].target_qubits) == {0, 2}
assert isinstance(tqueue[7], gates.CNOT)
assert tqueue[7].target_qubits == (0, )
assert isinstance(tqueue[8], gates.SWAP)
assert set(tqueue[8].target_qubits) == {0, 2}
assert isinstance(tqueue[9], gates.SWAP)
assert set(tqueue[9].target_qubits) == {1, 3}
qibo.set_backend(original_backend)
def test_ndevices():
"""Check that ``ndevices`` and ``nglobal`` is set properly."""
devices = {"/GPU:0": 2, "/GPU:1": 2}
c = models.DistributedCircuit(5, devices)
assert c.ndevices == 4
assert c.nglobal == 2
def test_invalid_devices():
"""Check if error is raised if total devices is not a power of 2."""
devices = {"/GPU:0": 2, "/GPU:1": 1}
with pytest.raises(ValueError):
c = models.DistributedCircuit(4, devices)
