Commit 3fed60c5 authored by Murali's avatar Murali
Browse files

Minor var name edit

parent 1a4265ce

func_matrix_vector.py

+2 −2
Original line number Diff line number Diff line
@@ -138,7 +138,7 @@ def Hele_Shaw(doc_id, args): 
    

    if args.savedata == True:

        MATRIX_SIZE = A_herm.shape[0]

        NUM_QUBITS = int(np.log2(MATRIX_SIZE))

        n_qubits_matrix = int(np.log2(MATRIX_SIZE))

        save_data = {'P_in'                  : P_in,

                     'P_out'                 : P_out,

                     'U_top'                 : U_top,
@@ -151,7 +151,7 @@ def Hele_Shaw(doc_id, args): 
                     'ny'                    : ny,

                     'A_herm'                : A_herm,

                     'B_herm'                : B_herm,

                     'NUM_QUBITS'            : NUM_QUBITS,

                     'n_qubits_matrix'       : n_qubits_matrix,

                     'args'                  : args}

        file = open(f'{filename}_metadata.pkl', 'wb')

        pickle.dump(save_data, file)

input_vars.yaml

+1 −1
Original line number Diff line number Diff line 
---

# Case 0: Sample system - tridiagonal matrix

case_name: Sample tridiagonal

NQ_MATRIX: 2       # Numer of qubits to determine size of linear system of quations being solved. A matrix size = 2^NUM_QUBITS.

NQ_MATRIX: 2       # Numer of qubits to determine size of linear system of quations being solved. A matrix size = 2^NQ_MATRIX.

savedir: models

savefilename: "{savedir}/sample_HHL"

test_linear_solver.py

+6 −6
Original line number Diff line number Diff line
@@ -18,7 +18,7 @@ import argparse 
import pickle



parser = argparse.ArgumentParser()

parser.add_argument("-nq", "--NUM_QUBITS", type=int, default=2, required=True, help="Numer of qubits to determine size of linear system of quations (A*x=b) being solved. Size of A matrix = 2^NUM_QUBITS.")

parser.add_argument("-nq", "--NQ_MATRIX", type=int, default=2, required=True, help="Numer of qubits to determine size of linear system of quations (A*x=b) being solved. Size of A matrix = 2^NQ_MATRIX.")

parser.add_argument("--savedata", default=False, action='store_true', help="Save data at `models/<filename>` with `<filename>` based on parameters.")

parser.add_argument("--gpu", default=False, action='store_true', help="Use GPU backend for Aer simulator.")

args = parser.parse_args()
@@ -32,8 +32,8 @@ args = parser.parse_args() 
#   vector = np.array([1, 0])

#   tridi_matrix = TridiagonalToeplitz(1, 1, -1 / 3)

# custom systems

NUM_QUBITS = args.NUM_QUBITS 

MATRIX_SIZE = 2 ** NUM_QUBITS

n_qubits_matrix = args.NQ_MATRIX 

MATRIX_SIZE = 2 ** n_qubits_matrix

# entries of the tridiagonal Toeplitz symmetric matrix

a = 1

b = -1/3
@@ -42,7 +42,7 @@ matrix = diags([b, a, b], 
               shape=(MATRIX_SIZE, MATRIX_SIZE)).toarray()

vector = np.array([1] + [0]*(MATRIX_SIZE - 1))

# we also generate an optimized matrix construction - tridiagonal toeplitz

tridi_matrix = TridiagonalToeplitz(NUM_QUBITS, a, b)

tridi_matrix = TridiagonalToeplitz(n_qubits_matrix, a, b)



# ============

# Select backend: Using different simulators (default in `linear_solvers` is statevector simulation)
@@ -143,7 +143,7 @@ print(f'diff (%): {np.abs(classical_solution.state-solvec_tridi)*100/classical_s 
savedata = args.savedata

if savedata == True:

    savedir = f'models'

    savefilename = f'{savedir}/sample_HHL_numq{NUM_QUBITS}_fulldata'

    savefilename = f'{savedir}/sample_HHL_numq{n_qubits_matrix}_fulldata'

    if not os.path.exists(savedir):

        os.mkdir(savedir)

    n=2
@@ -152,7 +152,7 @@ if savedata == True: 
        n+=1



    savefilename += '.pkl' 

    save_data = {   'NUM_QUBITS'            : NUM_QUBITS,

    save_data = {   'n_qubits_matrix'       : n_qubits_matrix,

                    'a'                     : a,

                    'b'                     : b,

                    'matrix'                : matrix,