def test_del_and_circuit(): with qp.Sim(enforce_release_at_zero=False, phase_fixup_bias=True): with qp.LogCirqCircuit() as circuit: with qp.qalloc(len=3, name='q') as q: q[0].clear(q[1] & q[2]) cirq.testing.assert_has_diagram(circuit, r""" q[0]: ---alloc---Mxc-------cxM---release--- | | q[1]: ---alloc---------@---------release--- | | | q[2]: ---alloc---------Z---------release--- """, use_unicode_characters=False) with qp.Sim(enforce_release_at_zero=False, phase_fixup_bias=False): with qp.LogCirqCircuit() as circuit: with qp.qalloc(len=3, name='b') as q: q[0].clear(q[1] & q[2]) cirq.testing.assert_has_diagram(circuit, r""" b[0]: ---alloc---Mxc---cxM---release--- | | b[1]: ---alloc---------------release--- | | b[2]: ---alloc---------------release--- """, use_unicode_characters=False)
def test_del_unary_circuit(): with qp.Sim(phase_fixup_bias=True): b = qp.qalloc(len=3, name='b') u = qp.qalloc(len=8, name='u') c = qp.qalloc(name='_c') with qp.LogCirqCircuit() as circuit: (1 << b).clear_storage_location(u, c) cirq.testing.assert_has_diagram(circuit, r""" _c: -------------------------------------------------------------------------------------------------------------------------------------------------------Z--------- b[0]: -------------------------------------------------------------------------------------------------------------------------------------Z------------------------- | b[1]: ---------------------------------------------------------------------------------------------Z-------------------Z-------------------|------------------------- | | | b[2]: -------------Z-------------------Z-------------------Z-------------------Z-------------------|-------------------|-------------------|------------------------- | | | | | | | u[0]: -------------|-------------------|-------------------|---------X---------@-------------------|---------X---------@---------X---------@---------Mxc-------cxM--- | | | | | | | u[1]: -------------|-------------------|---------X---------@---------|-------------------X---------@---------|-------------------@---Mxc-------cxM------------------- | | | | | | u[2]: -------------|---------X---------@---------|-------------------|-------------------|-------------------@---Mxc-------cxM--------------------------------------- | | | | | u[3]: ---X---------@---------|-------------------|-------------------|-------------------@---Mxc-------cxM----------------------------------------------------------- | | | | u[4]: ---|-------------------|-------------------|-------------------@---Mxc-------cxM------------------------------------------------------------------------------- | | | u[5]: ---|-------------------|-------------------@---Mxc-------cxM--------------------------------------------------------------------------------------------------- | | u[6]: ---|-------------------@---Mxc-------cxM----------------------------------------------------------------------------------------------------------------------- | u[7]: ---@---Mxc-------cxM------------------------------------------------------------------------------------------------------------------------------------------- """, use_unicode_characters=False) with qp.Sim(phase_fixup_bias=False): b = qp.qalloc(len=3, name='b') u = qp.qalloc(len=8, name='u') c = qp.qalloc(name='_c') with qp.LogCirqCircuit() as circuit: (1 << b).clear_storage_location(u, c) cirq.testing.assert_has_diagram(circuit, r""" u[0]: ---------------------------------------------------X-------------------------------X---------------X---------------Mxc---cxM--- | | | u[1]: -----------------------------------X---------------|---------------X---------------|---------------@---Mxc---cxM--------------- | | | | u[2]: -------------------X---------------|---------------|---------------|---------------@---Mxc---cxM------------------------------- | | | | u[3]: ---X---------------|---------------|---------------|---------------@---Mxc---cxM----------------------------------------------- | | | | u[4]: ---|---------------|---------------|---------------@---Mxc---cxM--------------------------------------------------------------- | | | u[5]: ---|---------------|---------------@---Mxc---cxM------------------------------------------------------------------------------- | | u[6]: ---|---------------@---Mxc---cxM----------------------------------------------------------------------------------------------- | u[7]: ---@---Mxc---cxM--------------------------------------------------------------------------------------------------------------- """, use_unicode_characters=False)
def test_optional(): def cf(x: qp.IntBuf, y: bool = True): x ^= int(y) @qp.semi_quantum(classical=cf) def qf(x: qp.Qubit, y: qp.Qubit.Borrowed = True): x ^= y b = qp.IntBuf.raw(val=0, length=1) qf.classical(b) assert int(b) == 1 qf.classical(b) assert int(b) == 0 qf.classical(b, True) assert int(b) == 1 qf.classical(b, False) assert int(b) == 1 with qp.Sim() as sim_state: q = qp.qalloc() assert not sim_state.resolve_location(q, False) qf(q) assert sim_state.resolve_location(q, False) qf(q) assert not sim_state.resolve_location(q, False) qf(q, True) assert sim_state.resolve_location(q, False) qf(q, False) assert sim_state.resolve_location(q, False) assert qp.measure(q, reset=True) qp.qfree(q)
def test_init_square_circuit(): with qp.Sim(phase_fixup_bias=True, enforce_release_at_zero=False): factor = qp.qalloc(len=2, name='f') out = qp.qalloc(len=4, name='s') with qp.LogCirqCircuit() as circuit: init_square(factor=factor, clean_out=out) cirq.testing.assert_has_diagram(circuit, r""" _do_addition_carry_in: -----------------------alloc---X-------@---------------------------------------------------------------@---@-------X---Mxc---cxM---release-----------------alloc---X-------@-------------------------------@---@-------X---Mxc---cxM---release------------- | | | | | | | | | | _sqr_offset: ---------------------alloc---X-----------|-------|---------------@---@---X---@---X-------@---@-------------------|---|-------|-------------------------X---release-----------|-------|-------------------------------|---|-------|----------------------------------- | | | | | | | | | | | | | | | | | | | _sqr_zero: ---------------alloc-----------|-----------|-------|---@---@---X---X---|---@---|---@---@---|---X---X-------@---@---|---|-------|-------------------------|---------------------|-------|---@---@---X---X-------@---@---|---|-------|-------------------------release--- | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | f[0]: ------------------------------------@-----------@---@---X---X---|---@-------|---|---|---|---|---|-------@---@---|---X---X---|---@---@-------------------------@---------------------|-------|---|---|---|---|-------|---|---|---|-------|----------------------------------- | | | | | | | | | | | | | | | | | | | | | | | | | | | | | f[1]: ------------------------------------@---------------|---|-------|---|-------|---|---|---|---|---|-------|---|---|-------|---|---|-----------------------------@---------------------@---@---X---X---|---@---@---@---|---X---X---|---@---@----------------------------------- | | | | | | | | | | | | | | | | | | | | | | | | | | s[0]: ----------------------------------------------------X---@-------|---|-------|---|---|---|---|---|-------|---|---|-------@---X---X-------------------------------------------------------|---|-------|---|---|---|---|-------|---|---|--------------------------------------- | | | | | | | | | | | | | | | | | | | | | s[1]: ----------------------------------------------------------------X---@-------|---|---|---|---|---|-------@---X---X-----------------------------------------------------------------------|---|-------|---|---|---|---|-------|---|---|--------------------------------------- | | | | | | | | | | | | | | | | s[2]: ----------------------------------------------------------------------------X---@---|---@---X---X---------------------------------------------------------------------------------------X---@-------|---|---|---|---|-------@---X---X--------------------------------------- | | | | | | s[3]: ------------------------------------------------------------------------------------X---------------------------------------------------------------------------------------------------------------X---@---@---X---X------------------------------------------------------- """, use_unicode_characters=False)
def _apply_classical(func: Callable, kwargs: Dict[str, Any]) -> Dict[str, Any]: kwargs = { k: v.copy() if hasattr(v, 'copy') else v for k, v in kwargs.items() } type_hints = get_type_hints(func) if 'control' in kwargs and 'control' not in type_hints: run = kwargs['control'] del kwargs['control'] else: run = True if run: if 'sim_state' in type_hints: kwargs['sim_state'] = qp.Sim() func(**kwargs) result = {'sim_state.phase_degrees': 0} for k, v in kwargs.items(): if isinstance(v, qp.IntBuf): result[k] = int(v) elif isinstance(v, qp.Sim): result['sim_state.phase_degrees'] = v.phase_degrees return result
def test_let_unary_circuit(): with qp.Sim(): b = qp.qalloc(len=3, name='b') u = qp.qalloc(len=8, name='u') c = qp.qalloc(name='_c') with qp.LogCirqCircuit() as circuit: (1 << b).init_storage_location(u, c) cirq.testing.assert_has_diagram(circuit, r""" _c: -----@----------------------------------------------------------- | b[0]: ---|---@------------------------------------------------------- | | b[1]: ---|---|-------@-------@--------------------------------------- | | | | b[2]: ---|---|-------|-------|-------@-------@-------@-------@------- | | | | | | | | u[0]: ---X---@---X---@---X---|-------@---X---|-------|-------|------- | | | | | | | | | | u[1]: -------X---@---|---|---@---X---|---|---@---X---|-------|------- | | | | | | | | | | u[2]: ---------------X---@---|---|---|---|---|---|---@---X---|------- | | | | | | | | | u[3]: -----------------------X---@---|---|---|---|---|---|---@---X--- | | | | | | | | u[4]: -------------------------------X---@---|---|---|---|---|---|--- | | | | | | u[5]: ---------------------------------------X---@---|---|---|---|--- | | | | u[6]: -----------------------------------------------X---@---|---|--- | | u[7]: -------------------------------------------------------X---@--- """, use_unicode_characters=False)
def test_plus_equal_gate_circuit(): with qp.Sim(enforce_release_at_zero=False): a = qp.qalloc(len=3, name='a') t = qp.qalloc(len=4, name='t') c = qp.qalloc(name='_c') with qp.LogCirqCircuit() as circuit: qp.arithmetic.do_addition(lvalue=t, offset=a, carry_in=c) cirq.testing.assert_has_diagram(circuit, r""" _c: -----X-------@---------------------------------------------------------------@---@-------X--- | | | | | a[0]: ---@---@---X---X-------@-----------------------------------@---@-------X---X---|---@---@--- | | | | | | | | | | a[1]: -------|---|---@---@---X---X-------@-------@---@-------X---X---|---@---@---|---|---|------- | | | | | | | | | | | | | | | a[2]: -------|---|-------|---|---@---@---X---@---X---|---@---@---|---|---|-------|---|---|------- | | | | | | | | | | | | | | | | t[0]: -------X---@-------|---|-------|---|---|---|---|---|-------|---|---|-------@---X---X------- | | | | | | | | | | | t[1]: -------------------X---@-------|---|---|---|---|---|-------@---X---X----------------------- | | | | | | t[2]: -------------------------------X---@---|---@---X---X--------------------------------------- | t[3]: ---------------------------------------X--------------------------------------------------- """, use_unicode_characters=False)
def test_if_less_than_then_circuit(): with qp.Sim(enforce_release_at_zero=False): lhs = qp.qalloc(len=4, name='lhs') rhs = qp.qalloc(len=4, name='rhs') c = qp.qalloc(name='_or_eq') t = qp.qalloc(name='t') with qp.LogCirqCircuit() as circuit: qp.arithmetic.do_if_less_than( lhs=lhs, rhs=rhs, or_equal=c, effect=t.__ixor__) cirq.testing.assert_has_diagram(circuit, r""" _or_eq: ---@---X---@-------------------------------------------------------------------------------@---X---@--- | | | | | | lhs[0]: ---X---|---@-------------------------------------------------------------------------------@---|---X--- | | | | lhs[1]: -------|---|---X-------@-------------------------------------------------------@-------X---|---|------- | | | | | | | | lhs[2]: -------|---|---|-------|---X-------@-------------------------------@-------X---|-------|---|---|------- | | | | | | | | | | | | lhs[3]: -------|---|---|-------|---|-------|---X-------@-------@-------X---|-------|---|-------|---|---|------- | | | | | | | | | | | | | | | | rhs[0]: -------@---X---@---X---@---|-------|---|-------|-------|-------|---|-------|---@---X---@---X---@------- | | | | | | | | | | | | rhs[1]: -------------------@---X---@---X---@---|-------|-------|-------|---@---X---@---X---@------------------- | | | | | | | | rhs[2]: -------------------------------@---X---@---X---@-------@---X---@---X---@------------------------------- | | | | rhs[3]: -------------------------------------------@---X---@---X---@------------------------------------------- | t: --------------------------------------------------------X--------------------------------------------------- """, use_unicode_characters=False)
def test_xor_equal_gate_circuit_2(): with qp.Sim(enforce_release_at_zero=False): with qp.LogCirqCircuit() as circuit: with qp.qalloc(len=3, name='a') as a: with qp.qalloc(len=4, name='t') as t: with qp.qalloc(name='_c') as c: t ^= a t ^= a & qp.controlled_by(c) cirq.testing.assert_has_diagram(circuit, r""" _c: ---------------------alloc---------------@---@---@---release----------------------- | | | a[0]: ---alloc-------------------@-----------@---|---|-----------------------release--- | | | | | | a[1]: ---alloc-------------------|---@-------|---@---|-----------------------release--- | | | | | | | a[2]: ---alloc-------------------|---|---@---|---|---@-----------------------release--- | | | | | | t[0]: -----------alloc-----------X---|---|---X---|---|-------------release------------- | | | | | | t[1]: -----------alloc---------------X---|-------X---|-------------release------------- | | | | t[2]: -----------alloc-------------------X-----------X-------------release------------- | | t[3]: -----------alloc---------------------------------------------release------------- """, use_unicode_characters=False)
def _apply_quantum(func: Callable, kwargs: Dict[str, Any]) -> Dict[str, Any]: with qp.Sim() as sim: type_hints = get_type_hints(func) mapped = {} for k, v in kwargs.items(): if type_hints[k] == qp.Quint: assert isinstance( v, qp.IntBuf ), 'Expected type qp.IntBuf for argument "{}"'.format(k) mapped[k] = qp.qalloc(len=len(v), name=k) mapped[k] ^= int(v) elif type_hints[k] == qp.Qubit: assert isinstance(v, qp.IntBuf) and len( v ) == 1, 'Expected length 1 qp.IntBuf for argument "{}"'.format( k) mapped[k] = qp.qalloc(name=k) mapped[k] ^= int(v) else: mapped[k] = v func(**mapped) result = {} for k, v in kwargs.items(): if type_hints[k] in [qp.Quint, qp.Qubit]: result[k] = qp.measure(mapped[k], reset=True) result['sim_state.phase_degrees'] = sim.phase_degrees return result
def test_eq(): with qp.Sim(): with qp.qalloc(len=4) as t: t.init(5) for k in range(-2, 60): assert qp.measure(t == k) == (k == 5) assert qp.measure(t, reset=True) == 5
def test_xor_equal_gate_circuit(): with qp.Sim(enforce_release_at_zero=False): with qp.LogCirqCircuit() as circuit: with qp.qalloc(len=3, name='a') as a: with qp.qalloc(len=4, name='t') as t: with qp.qalloc(name='_c') as c: qp.arithmetic.do_xor(lvalue=t, mask=a) qp.arithmetic.do_xor(lvalue=t, mask=a, control=c) cirq.testing.assert_has_diagram(circuit, r""" _c: ---------------------alloc---------------@---@---@---release----------------------- | | | a[0]: ---alloc-------------------@-----------@---|---|-----------------------release--- | | | | | | a[1]: ---alloc-------------------|---@-------|---@---|-----------------------release--- | | | | | | | a[2]: ---alloc-------------------|---|---@---|---|---@-----------------------release--- | | | | | | t[0]: -----------alloc-----------X---|---|---X---|---|-------------release------------- | | | | | | t[1]: -----------alloc---------------X---|-------X---|-------------release------------- | | | | t[2]: -----------alloc-------------------X-----------X-------------release------------- | | t[3]: -----------alloc---------------------------------------------release------------- """, use_unicode_characters=False)
def test_neq(): with qp.Sim(): with qp.qalloc(len=4) as t: t.init(5) for k in range(-2, 60): with qp.hold(t != k) as q: assert qp.measure(q) == (k != 5) assert qp.measure(t, reset=True) == 5
def test_uncompute(): for a, b, p in itertools.product([False, True], repeat=3): with qp.Sim(phase_fixup_bias=p): with qp.hold(a, name='a') as qa: with qp.hold(b, name='b') as qb: with qp.hold(a & b, name='c') as qc: assert qp.measure(qa) == a assert qp.measure(qb) == b assert qp.measure(qc) == (a and b)
def test_cmp(): with qp.Sim(): with qp.qalloc(len=4) as t: t.init(5) for k in range(-5, 30): assert qp.measure(t >= k) == (5 >= k) assert qp.measure(t > k) == (5 > k) assert qp.measure(t < k) == (5 < k) assert qp.measure(k < t) == (k < 5) assert qp.measure(t <= k) == (5 <= k) assert qp.measure(t, reset=True) == 5
def test_sim(): v1 = 15 v2 = 235 offset = 4 bits = 10 with qp.Sim(): with qp.hold(val=v1, name='a') as a: with qp.qalloc(len=bits, name='out') as out: out += a * v2 out += offset result = qp.measure(out, reset=True) assert result == (v1*v2 + offset) & ((1 << bits) - 1)
def test_integration(exp_len: int, modulus_len: int, emulate_additions: bool): with qp.Sim(emulate_additions=emulate_additions): exponent = random.randint(0, 1 << exp_len) modulus = random.randint(0, 1 << modulus_len) * 2 + 1 base = 0 while math.gcd(base, modulus) != 1 or base >= modulus: base = random.randint(0, 1 << modulus_len) * 2 + 1 with qp.hold(exponent) as exp: actual = measure_pow_mod(base=base, exponent=exp, modulus=modulus) expected = pow(base, exponent, modulus) assert actual == expected
def test_let_and_circuit(): with qp.Sim(enforce_release_at_zero=False): with qp.LogCirqCircuit() as circuit: with qp.qalloc(len=3, name='q') as q: q[0].init(q[1] & q[2]) cirq.testing.assert_has_diagram(circuit, r""" q[0]: ---alloc---X---release--- | | | q[1]: ---alloc---@---release--- | | | q[2]: ---alloc---@---release--- """, use_unicode_characters=False)
def test_with_sim_state(): def cf(sim_state: qp.ClassicalSimState): sim_state.phase_degrees += 180 @qp.semi_quantum(classical=cf) def qf(): qp.phase_flip() with qp.Sim() as sim_state: assert sim_state.phase_degrees == 0 qf() assert sim_state.phase_degrees == 180 qf.sim(sim_state) assert sim_state.phase_degrees == 0
def test_count(): v1 = 15 v2 = 235 offset = 4 bits = 100 with qp.Sim(): with qp.CountNots() as counts: with qp.hold(val=v1, name='a') as a: with qp.qalloc(len=bits, name='out') as out: out += a * v2 out += offset _ = qp.measure(out, reset=True) assert len(counts.keys()) == 3 assert counts[0] > 0 assert counts[1] > 0 assert 0 < counts[2] <= 1000
def test_xor_lookup(): with qp.Sim(phase_fixup_bias=True, enforce_release_at_zero=False): with qp.LogCirqCircuit() as circuit: with qp.qalloc(len=4, name='addr') as addr: with qp.qalloc(len=8, name='out') as out: with qp.qalloc(name='cnt') as cnt: out ^= qp.LookupTable(range( 1, 17))[addr] & qp.controlled_by(cnt) cirq.testing.assert_has_diagram(circuit, r""" _lookup_prefix: -----------------------------alloc---X---X-----------@---@-------------------------------------------------------------------------------------------------------------------------------------------------------------@-------------------------------------------------------------------------------------------------------------------------------------------------------------------@-------------------X-----------@---@-------------------------------------------------------------------------------------------------------------------------------------------------------------@-------------------------------------------------------------------------------------------------------------------------------------------------------------------@-------------------Mxc-------cxM---release--------------------------------- | | | | | | | | | | | _lookup_prefix_1: -----------------------------------|---|---alloc---X---X-----------@---@---------------------------------------------------------@---------------------------------------------------------------@-------------------X-----------@---@---------------------------------------------------------@---------------------------------------------------------------@-------------------Mxc---|---cxM---release---|---alloc---X---X-----------@---@---------------------------------------------------------@---------------------------------------------------------------@-------------------X-----------@---@---------------------------------------------------------@---------------------------------------------------------------@-------------------Mxc---|---cxM---release----------------------------------------------------------- | | | | | | | | | | | | | | | | | | | | | | | _lookup_prefix_2: -----------------------------------|---|-----------|-------alloc---X---X-----------@---@-------@-------------@-------------------X-----------@---@-------@-------------@-------------------Mxc---|---cxM---release-------alloc---X---X-----------@---@-------@-------------@-------------------X-----------@---@-------@-------------@-------------------Mxc---|---cxM---release---------|-------------------|-----------|-------alloc---X---X-----------@---@-------@-------------@-------------------X-----------@---@-------@-------------@-------------------Mxc---|---cxM---release-------alloc---X---X-----------@---@-------@-------------@-------------------X-----------@---@-------@-------------@-------------------Mxc---|---cxM---release---------|--------------------------------------------------------------------------- | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | _lookup_prefix_3: -----------------------------------|---|-----------|---------------|-------alloc---X---X---@---X---@---Mxc---|---cxM---release-------alloc---X---X---@---X---@---Mxc---|---cxM---release---------|-------------------------------|-------alloc---X---X---@---X---@---Mxc---|---cxM---release-------alloc---X---X---@---X---@---Mxc---|---cxM---release---------|-------------------------|-------------------|-----------|---------------|-------alloc---X---X---@---X---@---Mxc---|---cxM---release-------alloc---X---X---@---X---@---Mxc---|---cxM---release---------|-------------------------------|-------alloc---X---X---@---X---@---Mxc---|---cxM---release-------alloc---X---X---@---X---@---Mxc---|---cxM---release---------|-------------------------|--------------------------------------------------------------------------- | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | addr[0]: ------------alloc---------------------------|---|-----------|---------------|---------------@-------|-------|---------Z-------------------------------@-------|-------|---------Z-------------------------|-------------------------------|---------------@-------|-------|---------Z-------------------------------@-------|-------|---------Z-------------------------|-------------------------|-------------------|-----------|---------------|---------------@-------|-------|---------Z-------------------------------@-------|-------|---------Z-------------------------|-------------------------------|---------------@-------|-------|---------Z-------------------------------@-------|-------|---------Z-------------------------|-------------------------|-----------------------------------------------------------------release--- | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | addr[1]: ------------alloc---------------------------|---|-----------|---------------@-----------------------|-------|-------------------------------------------------|-------|-----------------------------------Z-------------------------------@-----------------------|-------|-------------------------------------------------|-------|-----------------------------------Z-------------------------|-------------------|-----------|---------------@-----------------------|-------|-------------------------------------------------|-------|-----------------------------------Z-------------------------------@-----------------------|-------|-------------------------------------------------|-------|-----------------------------------Z-------------------------|-----------------------------------------------------------------release--- | | | | | | | | | | | | | | | | | | | | | | | | | addr[2]: ------------alloc---------------------------|---|-----------@---------------------------------------|-------|-------------------------------------------------|-------|-------------------------------------------------------------------------------------------|-------|-------------------------------------------------|-------|-------------------------------------------------------------Z-------------------|-----------@---------------------------------------|-------|-------------------------------------------------|-------|-------------------------------------------------------------------------------------------|-------|-------------------------------------------------|-------|-------------------------------------------------------------Z-----------------------------------------------------------------release--- | | | | | | | | | | | | | | | | | | | | | addr[3]: ------------alloc---------------------------@---|---------------------------------------------------|-------|-------------------------------------------------|-------|-------------------------------------------------------------------------------------------|-------|-------------------------------------------------|-------|---------------------------------------------------------------------------------|---------------------------------------------------|-------|-------------------------------------------------|-------|-------------------------------------------------------------------------------------------|-------|-------------------------------------------------|-------|---------------------------------------------------------------------------------------Z---------------------------------------release--- | | | | | | | | | | | | | | | | | | | | cnt: --------------------------------alloc-----------@---@---------------------------------------------------|-------|-------------------------------------------------|-------|-------------------------------------------------------------------------------------------|-------|-------------------------------------------------|-------|---------------------------------------------------------------------------------@---------------------------------------------------|-------|-------------------------------------------------|-------|-------------------------------------------------------------------------------------------|-------|-------------------------------------------------|-------|---------------------------------------------------------------------------------------@-------------------release----------------------- | | | | | | | | | | | | | | | | out[0]: ---------------------alloc---------------------------------------------------------------------------X-------|-------------------------------------------------X-------|-------------------------------------------------------------------------------------------X-------|-------------------------------------------------X-------|-------------------------------------------------------------------------------------------------------------------------------------X-------|-------------------------------------------------X-------|-------------------------------------------------------------------------------------------X-------|-------------------------------------------------X-------|---------------------------------------------------------------------------------------------------------------------release------------- | | | | | | | | | | | | | | | | | out[1]: ---------------------alloc-----------------------------------------------------------------------------------X-------------------------------------------------X-------|-------------------------------------------------------------------------------------------|-------X-------------------------------------------------X-------|-------------------------------------------------------------------------------------------------------------------------------------|-------X-------------------------------------------------X-------|-------------------------------------------------------------------------------------------|-------X-------------------------------------------------X-------|---------------------------------------------------------------------------------------------------------------------release------------- | | | | | | | | | | | | | | | out[2]: ---------------------alloc---------------------------------------------------------------------------------------------------------------------------------------------X-------------------------------------------------------------------------------------------X-------X-------------------------------------------------X-------|-------------------------------------------------------------------------------------------------------------------------------------|-------|-------------------------------------------------|-------X-------------------------------------------------------------------------------------------X-------X-------------------------------------------------X-------|---------------------------------------------------------------------------------------------------------------------release------------- | | | | | | | | | | | out[3]: ---------------------alloc-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------X-------------------------------------------------------------------------------------------------------------------------------------X-------X-------------------------------------------------X-------X-------------------------------------------------------------------------------------------X-------X-------------------------------------------------X-------|---------------------------------------------------------------------------------------------------------------------release------------- | | | out[4]: ---------------------alloc---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------X---------------------------------------------------------------------------------------------------------------------release------------- | | out[5]: ---------------------alloc-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------release------------- | | out[6]: ---------------------alloc-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------release------------- | | out[7]: ---------------------alloc-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------release------------- """, use_unicode_characters=False)
def sim_call(func: Callable, *args, **kwargs) -> 'qp.ArgsAndKwargs': def qalloc_as_needed(a: qp.ArgParameter): if a.parameter_type is None: raise ValueError('Function arguments must be annotated, so that ' 'quantum values can be allocated if needed.') if a.parameter_type is qp.Quint: if isinstance(a.arg, qp.IntBuf): n = len(a.arg) elif isinstance(a.arg, int): n = a.arg.bit_length() else: raise ValueError('Unsupported Quint input: {}'.format(a)) result = qp.qalloc(len=n, name=a.parameter.name) result.init(a.arg) return result if a.parameter_type is qp.QuintMod: assert isinstance(a.arg, ModInt) result = qp.qalloc(modulus=a.arg.modulus, name=a.parameter.name) result.init(a.arg.val) return result if a.parameter_type is qp.Qubit: result = qp.qalloc(name=a.parameter.name) result.init(a.arg) return result return a.arg def qfree_as_needed(a: Any): if isinstance(a, (qp.Quint, qp.Qureg, qp.Qubit, qp.QuintMod)): result = qp.measure(a, reset=True) qp.qfree(a) return result return a matched = qp.ArgsAndKwargs(args, kwargs).match_parameters(func) with qp.Sim(): allocated = matched.map(qalloc_as_needed) allocated.pass_into(func) return allocated.map(qfree_as_needed)
def test_eq_circuit(): with qp.Sim(enforce_release_at_zero=False, phase_fixup_bias=True): with qp.LogCirqCircuit() as circuit: with qp.qalloc(len=4, name='lhs') as lhs: with qp.hold(lhs == 5, name='target'): pass cirq.testing.assert_has_diagram(circuit, r""" lhs[0]: ---alloc---------------@-----------------@-----------------------release--- | | | | lhs[1]: ---alloc-----------X---@---X---------X---@---X-------------------release--- | | | | | | | | lhs[2]: ---alloc-----------|---@---|---------|---@---|-------------------release--- | | | | | | | | lhs[3]: ---alloc-----------X---@---X---------X---Z---X-------------------release--- | target: -----------alloc-------X-------Mxc---------------cxM---release------------- """, use_unicode_characters=False)
def test_classical(): def g(x: qp.IntBuf, y: int): assert isinstance(x, qp.IntBuf) assert isinstance(y, int) x ^= y @qp.semi_quantum(classical=g) def f(x: qp.Quint, y: qp.Quint.Borrowed): x ^= y assert f.classical is g with qp.Sim() as sim_state: q = qp.qalloc(len=5) assert sim_state.resolve_location(q, False) == 0 f.sim(sim_state, q, 3) assert sim_state.resolve_location(q, False) == 3 f.sim(sim_state, q, 5) assert sim_state.resolve_location(q, False) == 6 f.sim(sim_state, q, 6) qp.qfree(q)
def test_xor_unary_circuit(): with qp.Sim(phase_fixup_bias=True): b = qp.qalloc(len=3, name='b') u = qp.qalloc(len=8, name='u') c = qp.qalloc(name='_c') with qp.LogCirqCircuit() as circuit: u ^= (1 << b) & qp.controlled_by(c) cirq.testing.assert_has_diagram(circuit, r""" _c: -------------------------@---@-------------------------------------------------------------------------------------------------------------------------------------------------------------@-------------------------------------------------------------------------------------------------------------------------------------------------------------------@------------------- | | | | _lookup_prefix: -----alloc---X---X-----------@---@---------------------------------------------------------@---------------------------------------------------------------@-------------------X-----------@---@---------------------------------------------------------@---------------------------------------------------------------@-------------------Mxc---|---cxM---release--- | | | | | | | | | | _lookup_prefix_1: -----------|-------alloc---X---X-----------@---@-------@-------------@-------------------X-----------@---@-------@-------------@-------------------Mxc---|---cxM---release-------alloc---X---X-----------@---@-------@-------------@-------------------X-----------@---@-------@-------------@-------------------Mxc---|---cxM---release---------|------------------- | | | | | | | | | | | | | | | | | | | | | | _lookup_prefix_2: -----------|---------------|-------alloc---X---X---@---X---@---Mxc---|---cxM---release-------alloc---X---X---@---X---@---Mxc---|---cxM---release---------|-------------------------------|-------alloc---X---X---@---X---@---Mxc---|---cxM---release-------alloc---X---X---@---X---@---Mxc---|---cxM---release---------|-------------------------|------------------- | | | | | | | | | | | | | | | | | | | | | | b[0]: -----------------------|---------------|---------------@-------|-------|---------Z-------------------------------@-------|-------|---------Z-------------------------|-------------------------------|---------------@-------|-------|---------Z-------------------------------@-------|-------|---------Z-------------------------|-------------------------|------------------- | | | | | | | | | | | | | | b[1]: -----------------------|---------------@-----------------------|-------|-------------------------------------------------|-------|-----------------------------------Z-------------------------------@-----------------------|-------|-------------------------------------------------|-------|-----------------------------------Z-------------------------|------------------- | | | | | | | | | | b[2]: -----------------------@---------------------------------------|-------|-------------------------------------------------|-------|-------------------------------------------------------------------------------------------|-------|-------------------------------------------------|-------|-------------------------------------------------------------Z------------------- | | | | | | | | u[0]: ---------------------------------------------------------------X-------|-------------------------------------------------|-------|-------------------------------------------------------------------------------------------|-------|-------------------------------------------------|-------|--------------------------------------------------------------------------------- | | | | | | | u[1]: -----------------------------------------------------------------------X-------------------------------------------------|-------|-------------------------------------------------------------------------------------------|-------|-------------------------------------------------|-------|--------------------------------------------------------------------------------- | | | | | | u[2]: -------------------------------------------------------------------------------------------------------------------------X-------|-------------------------------------------------------------------------------------------|-------|-------------------------------------------------|-------|--------------------------------------------------------------------------------- | | | | | u[3]: ---------------------------------------------------------------------------------------------------------------------------------X-------------------------------------------------------------------------------------------|-------|-------------------------------------------------|-------|--------------------------------------------------------------------------------- | | | | u[4]: -----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------X-------|-------------------------------------------------|-------|--------------------------------------------------------------------------------- | | | u[5]: -------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------X-------------------------------------------------|-------|--------------------------------------------------------------------------------- | | u[6]: ---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------X-------|--------------------------------------------------------------------------------- | u[7]: -----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------X--------------------------------------------------------------------------------- """, use_unicode_characters=False)
def test_redundant_lookup(): with qp.Sim(phase_fixup_bias=True, enforce_release_at_zero=False): with qp.LogCirqCircuit() as circuit: with qp.qalloc(len=4, name='addr') as addr: with qp.qalloc(len=8, name='out') as out: with qp.qalloc(name='cnt') as cnt: out ^= qp.LookupTable( [3] * 16)[addr] & qp.controlled_by(cnt) cirq.testing.assert_has_diagram(circuit, r""" addr[0]: ---alloc-------------------------------------------release--- | | addr[1]: ---alloc-------------------------------------------release--- | | addr[2]: ---alloc-------------------------------------------release--- | | addr[3]: ---alloc-------------------------------------------release--- cnt: -----------------------alloc---@---release----------------------- | out[0]: ------------alloc-----------X-------------release------------- | | | out[1]: ------------alloc-----------X-------------release------------- | | out[2]: ------------alloc-------------------------release------------- | | out[3]: ------------alloc-------------------------release------------- | | out[4]: ------------alloc-------------------------release------------- | | out[5]: ------------alloc-------------------------release------------- | | out[6]: ------------alloc-------------------------release------------- | | out[7]: ------------alloc-------------------------release------------- """, use_unicode_characters=False)
def test_del_lookup(): with qp.Sim(phase_fixup_bias=True, enforce_release_at_zero=False): with qp.LogCirqCircuit() as circuit: with qp.qalloc(len=4, name='addr') as addr: with qp.qalloc(name='cnt'): with qp.hold(qp.LookupTable(range(1, 17))[addr], name='out'): circuit[:] = [] cirq.testing.assert_has_diagram(circuit, r""" _lookup_prefix: ---------------------------------------alloc---X---X-----------@---@---------------@-----------------@-------------------X-----------@---@-------@---------------------@-------------------Mxc-------cxM---release-------------------------------------------------------------------------------------------------------------------- | | | | | | | | | _lookup_prefix_1: ---------------------------------------------|-------alloc---X---X---@---@---@---X---@---@---Mxc---|---cxM---release-------alloc---X---X---@---X---@---@---@---Mxc---|---cxM---release---------------------------------------------------------------------------------------------------------------------------------------------- | | | | | | | | | | | | | | addr[0]: ----------------------@-------------------------------|---------------|-------|---|---|-------|---|---------|-------------------------------|-------|-------|---|---|---------|-----------------------------------------------------------------------------------------------Z----------------------------------------------------release--- | | | | | | | | | | | | | | | | | addr[1]: ----------------------|-------@-------@---------------|---------------|-------|---|---|-------|---|---------|-------------------------------|-------|-------|---|---|---------|-------------------------------------------------------Z-------------------Z-------------------|----------------------------------------------------release--- | | | | | | | | | | | | | | | | | | | | | addr[2]: ----------------------|-------|-------|---------------|---------------@-------|---|---|-------|---|---------Z-------------------------------@-------|-------|---|---|---------Z-------------------------------------------------------|-------------------|-------------------|----------------------------------------------------release--- | | | | | | | | | | | | | | | | | addr[3]: ----------------------|-------|-------|---------------@-----------------------|---|---|-------|---|-------------------------------------------------|-------|---|---|-----------------------------------Z-----------------------------|-------------------|-------------------|----------------------------------------------------release--- | | | | | | | | | | | | | | | cnt: --------------------------|-------|-------|---------------------------------------|---|---|-------|---|-------------------------------------------------|-------|---|---|-----------------------------------------------------------------|-------------------|-------------------|------------------------------------------release------------- | | | | | | | | | | | | | | | out[0]: -------------Mxc---X---@---X---@---X---|---------------------------------------Z---|---|-------|---|-------------------------------------------------|-------Z---|---|-----------------------------------------------------------------|---------X---------@---------X---------@---------Mxc--------cxM---cxM---release----------------------- | | | | | | | | | | | | | | | | | | out[1]: -------------Mxc-------X---@---|---|---@---X---------------------------------------Z---|-------|---|-------------------------------------------------|-----------Z---|-------------------------------------------------------X---------@---------|-------------------@---Mxc-------cxM--------------------cxM---release----------------------- | | | | | | | | | | | | | | out[2]: -------------Mxc---------------X---@---|---|-------------------------------------------|-------Z---|-------------------------------------------------Z---------------|-------------------------------------------------------|-------------------@---Mxc-------cxM----------------------------------------cxM---release----------------------- | | | | | | | | | out[3]: -------------Mxc-----------------------X---@-------------------------------------------Z-----------Z-----------------------------------------------------------------Z-------------------------------------------------------@---Mxc-------cxM------------------------------------------------------------cxM---release----------------------- | | | out[4]: -------------Mxc------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------cxM---release----------------------- global phase: pi """, use_unicode_characters=False)