mirror of https://github.com/intel/intel-qs.git
68 lines
2.4 KiB
Python
68 lines
2.4 KiB
Python
import sys
|
|
sys.path.insert(0, '../build/lib')
|
|
import intelqs_py as iqs
|
|
import numpy as np
|
|
|
|
|
|
def run_circuit(num_qubits):
|
|
reg = iqs.QubitRegister(num_qubits, 'base', 0, 0)
|
|
for i in range(num_qubits):
|
|
reg.ApplyHadamard(i)
|
|
reg.ApplyRotationZ(i, np.pi/3)
|
|
return reg
|
|
|
|
|
|
if __name__ == '__main__':
|
|
if len(sys.argv) <= 1:
|
|
print('Usage: {} <num_qubits>'.format(sys.argv[0]))
|
|
sys.exit(0)
|
|
num_qubits = int(sys.argv[1])
|
|
|
|
iqs.EnvInit()
|
|
if iqs.MPIEnvironment.IsUsefulRank()==False:
|
|
iqs.EnvFinalize()
|
|
exit()
|
|
# The simulation of a N-qubit system cannot be divided in more than 2^(N-1) ranks.
|
|
if iqs.MPIEnvironment.GetStateSize()>2**(num_qubits-1) or num_qubits<1:
|
|
if iqs.MPIEnvironment.GetRank()==0:
|
|
print("No more than 2^(N-1) useful ranks for a N-qubit state.")
|
|
iqs.EnvFinalize()
|
|
exit()
|
|
|
|
reg = run_circuit(num_qubits)
|
|
state_vector = np.array(reg, copy=False)
|
|
|
|
rank = iqs.MPIEnvironment.GetRank()
|
|
if num_qubits<7:
|
|
print('\nFinal state at rank {}: {}'.format(rank, state_vector))
|
|
elif rank==0:
|
|
print('\nState too large to be printed, try 4 qubits :-)\n')
|
|
|
|
iqs.EnvFinalize()
|
|
|
|
# SAMPLE RUNS:
|
|
|
|
## Single-Process:
|
|
'''
|
|
python python_mpi.py 4
|
|
[|s0>: 0] world_rank: 0 , state_rank: 0 (state 0 of 1) my_node_id: 0 , num_nodes: 1 , ranks/node: 1 , threads/rank: 72 --useful
|
|
|
|
Final state at rank 0: [0.21650635-0.125j 0.21650635-0.125j 0.21650635-0.125j 0.21650635-0.125j
|
|
0.21650635-0.125j 0.21650635-0.125j 0.21650635-0.125j 0.21650635-0.125j
|
|
0.21650635+0.125j 0.21650635+0.125j 0.21650635+0.125j 0.21650635+0.125j
|
|
0.21650635+0.125j 0.21650635+0.125j 0.21650635+0.125j 0.21650635+0.125j]
|
|
'''
|
|
|
|
## Multi-Process:
|
|
'''
|
|
mpiexec.hydra -n 2 python python_mpi.py 4
|
|
[|s0>: 0] world_rank: 0 , state_rank: 0 (state 0 of 1) my_node_id: 0 , num_nodes: 1 , ranks/node: 2 , threads/rank: 36 --useful
|
|
[|s0>: 1] world_rank: 1 , state_rank: 1 (state 0 of 1) my_node_id: 0 , num_nodes: 1 , ranks/node: 2 , threads/rank: 36 --useful
|
|
|
|
Final state at rank 0: [0.21650635-0.125j 0.21650635-0.125j 0.21650635-0.125j 0.21650635-0.125j
|
|
0.21650635-0.125j 0.21650635-0.125j 0.21650635-0.125j 0.21650635-0.125j]
|
|
|
|
Final state at rank 1: [0.21650635+0.125j 0.21650635+0.125j 0.21650635+0.125j 0.21650635+0.125j
|
|
0.21650635+0.125j 0.21650635+0.125j 0.21650635+0.125j 0.21650635+0.125j]
|
|
'''
|