Skip to content

Grover from functional building blocks

View on GitHub Experiment in the IDE

Setting the scene

# !pip install -U classiq

new_classiq_user = False
if new_classiq_user:
    import classiq

    classiq.authenticate()

Warm Up

First Example

Write a function that prepares the minus state \(\ket{-}=\frac{1}{\sqrt2}(\ket{0}-\ket{1})\), assuming it recives the qubit \(\ket{x}=\ket{0}\) (hint:

HINT Use `H(x)`,`X(x)` 
from classiq import H, QBit, X, qfunc


@qfunc
def prepare_minus_state(x: QBit):
    pass
    # TODO complete here

Now we will test our code:

from classiq import Output, allocate


@qfunc
def main(x: Output[QBit]):
    allocate(1, x)
    prepare_minus_state(x)  # Prepare the minus state

from classiq import create_model, synthesize

quantum_model = create_model(main)
quantum_program = synthesize(quantum_model)

from classiq import show

show(quantum_program)

Opening: https://platform.classiq.io/circuit/1bb9494d-6aa4-48e2-9596-3f39debea774?version=0.41.0.dev39%2B79c8fd0855

Some basic explanations about the high-level functional design with Classiq:

  • There should always be a main (def main(...)) function - the model that captures your algortihm is described there

  • The model is always generated out of the main function

  • The model is sent to the synthesis engine (compiler) that return a quantum program which contains the quantum circuit

Some basic guidelines about the modeling language (QMOD):

  1. Every quantum variable should be declared, either as a parameter of a funciton e.g. def prepare_minus(x: QBit) or within the function itself with x = QBit('x')

  2. Some quantum variables need to be initalized with the allocate function. This is required in 2 cases:

  3. A variable is a parameter of a function with the declaration Output like def main(x: Output[QNum])

  4. A variable that was declared within a function like a = QNum('a')

  5. For the main function, you will always use Output for all variables, as the function does not receive any input

Important tip!

You can see all the declarations of the functions with what are their input arguments in the functions.py file within the classiq package (or by just right clicking a function and presing Go To Defintion)

Uniform Superposition

Let's continue warming up with creating a function that receives a quantum register and creates a uniform superposition for all qubits within this array. You should use the function apply_to_all(gate_operand=, target=):

from classiq import QArray, apply_to_all


@qfunc
def create_initial_state(reg: QArray[QBit]):
    pass
    # TODO complete here apply_to_all(gate_operand=, target=)

Test yout function by creating a new main function, synthesizing and viewing the circuit:

@qfunc
def main(x: Output[QArray[QBit]]):
    allocate(7, x)
    create_initial_state(x)

model = create_model(main)
qprog = synthesize(model)
show(qprog)

Opening: https://platform.classiq.io/circuit/6ee14d77-ba40-4733-9a2d-0740f4390b6a?version=0.41.0.dev39%2B79c8fd0855

Function of a function

In our Grover example we will have 3 variables a,b,c. We want to prepare all of them in an initial state of equal superposition. Create a fucntion that receives these 3 quantum variables as quantum integers (QNum) and applies the create_inital_state function to each:

from classiq import QNum


@qfunc
def create_initial_states(a: QNum, b: QNum, c: QNum):
    pass
    # TODO Complete here

You can create a main function, synthesize and visualize the generated circuit if you want to test yourself.

Oracle - Reflection around bad states

Tehoretical Background

Overall we can understand the Grover operator as composed of two reflection operators: 1. Around the superposition of 'bad states' (i.e. not the solutions) 2. Around the initial guess state

In this section we will build the first reflection operator which is also the implementation of the oracle function. Geometrically it can be understood in the 2D vector space of \(span\{\ket{\psi_{good}},\ket{\psi_{bad}}\}\).

Oracle

The above figures describes geometrically the reflection of some state \(\ket{\psi}=\alpha\ket{\psi_{good}}+\beta\ket{\psi_{bad}}\) around the state \(\ket{\psi_{bad}}\) such that \(\begin{equation} R(\alpha\ket{\psi_{good}}+\beta\ket{\psi_{bad}}) = -\alpha\ket{\psi_{good}}+\beta\ket{\psi_{bad}} \end{equation}\)

This operator can also be written as \(\begin{equation} R\ket{x}=(-1)^{x==\text{good solution}}\ket{x} \end{equation}\)

so if the state of \(x\) is a solution it gets a \((-)\) phase.

Implementation

Now we will actually implement the black box, the oracle, people are speaking about in quantum algortihms. The beuty of Classiq is that we just need to specify its functionality, and Classiq automatically implements it for us. For our purposes, we want to find all the states that obey \(2a+b=c\) so there are 3 quantum variables. In addition, we want to store our results somewhere, i.e. to indicate for each tupple of 3 nunmbers (a,b,c) (e.g. (1,2,2)) if the state is what we are looking for (\(2*1+2=4!=2\) so the result is FALSE in this case).

We will store the result in the variable res such that: \(\begin{equation} res = res \oplus (2a+b==c) \end{equation}\)

What we really want to implement here is graphically described as:

res

Adapt the following function so it will apply the desired equation: 

@qfunc
def oracle_black_box(res: QNum, a: QNum, b: QNum, c: QNum):
    # TODO Adapt with the correct statement
    res ^= a + b + c == 8

Now let's go quantum! We want to store the result of the above operation in the phase of the state \(\ket{a,b,c}\). That is, we want \(\begin{equation} \ket{a,b,c}\rightarrow(-1)^{(2a+b==c)}\ket{a,b,c}\end{equation}\)

There is a common procedure in quantum algortihm that applies this and it is called phase kickback. It's working by applying the above oracle_black_box function to an initial res qubit to the state \(\ket{-}\) (has anyone prepared a function prepare_minus(x) by any chance?)

You can work out the math that the following scheme imoplement what we want:

phase_kickback_scheme

Now we can implement it using our oracle:

from classiq import free, invert


@qfunc
def oracle_function(a: QNum, b: QNum, c: QNum):
    aux = QBit("aux")

    allocate(1, aux)
    prepare_minus_state(aux)

    oracle_black_box(aux, a, b, c)

    # We want to bring the aux state back to it's initial value and to free it up for further use
    invert(lambda: prepare_minus_state(aux))
    free(aux)  # and that it can be re-used

Diffuser - Reflection around initial guess

Theoretical Background

The second part of the Grover operator is the diffuser, which can be viewed as the reflection operator around our initial guess.

diffuser

As with the oracle reflection operator, we can describe any state \(\ket{psi}\) as a superposition of the initial state \(\ket{\psi_0}\) such that and the orthogoanl state to it \(\ket{\psi_0^{\bot}}\)

\(\begin{equation} \ket{\psi} = \alpha \ket{\psi_0} +\beta \ket{\psi_0^{\bot}} \end{equation}\)

Here what we want to implement is to add a \((-)\) phase for all states that are not equal our initial guess, that is the reflection operator (our diffuser) is defined as:

\(\begin{equation} R(\alpha \ket{\psi_0} +\beta \ket{\psi_0^{\bot}}) = \alpha \ket{\psi_0} -\beta \ket{\psi_0^{\bot}} \end{equation}\)

In order to implement the reflection around our initial state, we will implement a reflection around the zero state \(\ket{0}\), and then squiz it between to state preperations operators for our initial state \(\ket{\psi_0}\). That is, if \(U_{\psi_0}\ket{0}=\ket{\psi_0}\) then we will implement the desired \(R\) operator with: \(\begin{equation} R = U_{\psi_0}R_0 U_{\psi_0}^{\dagger} \end{equation}\)

where \(R_0\) is the reflection operator around the zero state: \(\begin{equation} R_0\ket{x} = (-1)^{x\ne0}\ket{x}= (2\ket{0}\bra{0}-I)\ket{x} \end{equation}\)

Implementation

First we will implement the not equal zero function which takes aux and x as inputs and applies \(\begin{equation} res = res \oplus (x\ne0) \end{equation}\)

@qfunc
def not_equal_zero(aux: QBit, x: QNum):
    aux ^= x == 0
    X(aux)

Now we will use the common trick of phase kick back again. As the aux qubit for the not_equal_zero funciton we need to insert the \(\ket{-}\) state after it has been initialized and apploied, and after the application of the function we need to return aux to its initial state and to free it up (We did this precedure before).

HINT 1. declare the auxilary qubit 2. initalize it using `allocate` 3. prepare the $\ket{-}$ state 4. Apply the `not_equal_zero` funciton 5. reverse the $\ket{-}$ with the `invert` operation 6. free the auxilary qubit 
@qfunc
def zero_diffuser(x: QNum):
    pass
    # TODO complete here

Now after we've implemented the zero diffuser, we need to sandwitch it with the state preperations of our initial state a,b,c. The tricky part here is that the zero diffuser expects to receive only 1 quantunm variable x but we have three. So what should we do? We combine them to one quantum variable with the bind operation, treating them as one variable, and then splitting them back into 3 variables with the bind operation again.

from classiq import bind

size_a = 2
size_b = 2
size_c = 3


@qfunc
def initial_state_diffuser(a: QNum, b: QNum, c: QNum):

    create_initial_states(a, b, c)

    abc = QNum("abc")
    bind([a, b, c], abc)
    zero_diffuser(abc)
    bind(abc, [a, b, c])
    invert(lambda: create_initial_states(a, b, c))

Putting all together

That's it! You've made it! Now is the time to harvest the fruits of the hard work and put everything together! Complete your grover operator by implementing the two functions that you've built, first the oracle_fucntion and then the inital_state_diffuser:

@qfunc
def my_grover_operator(a: QNum, b: QNum, c: QNum):
    # TODO complete here
    pass

Now that we have our Grover operator, we can run it within our code. We have 3 steps here: 1. initalize a,b and c within the scope of the main function using the allocate operation 2. create the initial states for a,b and c 3. Apply your Grover operator

@qfunc
def main(a: Output[QNum], b: Output[QNum], c: Output[QNum]):
    allocate(size_a, a)
    allocate(size_b, b)
    allocate(size_c, c)

    # TODO complete here
    pass

Synthesize your model:

qmod = create_model(main)
qprog = synthesize(qmod)

And view it within the IDE:

show(qprog)

Opening: https://platform.classiq.io/circuit/40c8f298-1e9e-4d9d-8c45-bb2d127444d6?version=0.41.0.dev39%2B79c8fd0855

Is it what you were expecting? Now we can play with the constraints as we did in the IDE:

from classiq import Constraints, set_constraints

qmod = set_constraints(qmod, Constraints(optimization_parameter="depth"))  # or 'width'
qprog = synthesize(qmod)
show(qprog)

Opening: https://platform.classiq.io/circuit/7916e73f-7c7c-4a75-bea7-915dd9438baa?version=0.41.0.dev39%2B79c8fd0855

CONGRADULATIONS!

You have completed your own Grover algortihm implementation from functional building blocks without swegging under the rug any details, really impressive work!

The full solution for your reference

from classiq import (
    Constraints,
    H,
    Output,
    QArray,
    QBit,
    QNum,
    X,
    allocate,
    apply_to_all,
    bind,
    create_model,
    free,
    invert,
    qfunc,
    set_constraints,
    show,
    synthesize,
    write_qmod,
)


@qfunc
def prepare_minus_state(x: QBit):
    X(x)
    H(x)


@qfunc
def create_initial_state(reg: QArray[QBit]):
    apply_to_all(gate_operand=H, target=reg)


@qfunc
def create_initial_state(reg: QArray[QBit]):
    apply_to_all(lambda qb: H(qb), reg)


@qfunc
def create_initial_states(a: QNum, b: QNum, c: QNum):
    create_initial_state(a)
    create_initial_state(b)
    create_initial_state(c)


@qfunc
def oracle_black_box(res: QNum, a: QNum, b: QNum, c: QNum):
    res ^= 2 * a + b == c


@qfunc
def oracle_function(a: QNum, b: QNum, c: QNum):
    aux = QBit("aux")

    allocate(1, aux)
    prepare_minus_state(aux)

    oracle_black_box(aux, a, b, c)

    # We want to bring the aux state back to it's initial value and to free it up for further use
    invert(lambda: prepare_minus_state(aux))
    free(aux)  # and that it can be re-used


@qfunc
def not_equal_zero(aux: QBit, x: QNum):
    aux ^= x == 0
    X(aux)


@qfunc
def zero_diffuser(x: QNum):
    aux = QBit("aux")
    allocate(1, aux)

    prepare_minus_state(aux)
    not_equal_zero(aux, x)
    invert(lambda: prepare_minus_state(aux))
    free(aux)


size_a = 2
size_b = 2
size_c = 3


@qfunc
def initial_state_diffuser(a: QNum, b: QNum, c: QNum):

    create_initial_states(a, b, c)

    abc = QNum("abc", size=size_a + size_b + size_c, fraction_digits=0, is_signed=False)
    bind([a, b, c], abc)
    zero_diffuser(abc)
    bind(abc, [a, b, c])
    invert(lambda: create_initial_states(a, b, c))


@qfunc
def my_grover_operator(a: QNum, b: QNum, c: QNum):
    oracle_function(a, b, c)
    initial_state_diffuser(a, b, c)


@qfunc
def main(a: Output[QNum], b: Output[QNum], c: Output[QNum]):
    allocate(size_a, a)
    allocate(size_b, b)
    allocate(size_c, c)
    create_initial_states(a, b, c)
    my_grover_operator(a, b, c)


qmod = create_model(main)
qprog = synthesize(qmod)
show(qprog)

qmod = set_constraints(qmod, Constraints(optimization_parameter="depth"))  # or 'width'
qprog = synthesize(qmod)
show(qprog)

write_qmod(qmod, "grover_workshop")

Opening: https://platform.classiq.io/circuit/7be18940-1e20-488e-88cb-7e66a2da8bef?version=0.41.0.dev39%2B79c8fd0855


Opening: https://platform.classiq.io/circuit/400603b7-92d0-4e5a-be8e-f9aef5897a1a?version=0.41.0.dev39%2B79c8fd0855