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)