Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

feat: qfasm support parameter #151

Merged
merged 1 commit into from
Mar 29, 2024
Merged

feat: qfasm support parameter #151

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions quafu/elements/utils.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -55,14 +55,14 @@ def handle_expression(param: ParameterType):
retstr = f"({retstr} - {handle_expression(param.operands[i])})"
elif param.funcs[i] == _operator.truediv:
retstr = f"{retstr} / {handle_expression(param.operands[i])}"
elif param.funcs[i] == _operator.pow:
retstr = f"({retstr}) ^ {handle_expression(param.operands[i])}"
elif param.funcs[i] == anp.sin:
retstr = f"sin({retstr})"
elif param.funcs[i] == anp.cos:
retstr = f"cos({retstr})"
elif param.funcs[i] == anp.tan:
retstr = f"tan({retstr})"
elif param.funcs[i] == _operator.pow:
retstr = f"pow({retstr}, {handle_expression(param.operands[i])})"
elif param.funcs[i] == anp.arcsin:
retstr = f"asin({retstr})"
elif param.funcs[i] == anp.arccos:
Expand Down
5 changes: 5 additions & 0 deletions quafu/qfasm/qfasm_lexer.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -78,6 +78,7 @@ def token(self):
"STRING",
"ASSIGN",
"MATCHES",
"EQUAL",
"ID",
"UNIT",
"CHANNEL",
Expand Down Expand Up @@ -147,6 +148,10 @@ def t_MATCHES(self, t):
r"=="
return t

def t_EQUAL(self, t):
r"="
return t

def t_UNIT(self, t):
r"ns|us"
return t
Expand Down
121 changes: 74 additions & 47 deletions quafu/qfasm/qfasm_parser.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,11 +23,12 @@
from quafu.qfasm.exceptions import ParserError

from quafu import QuantumCircuit

from quafu.elements import Parameter, ParameterExpression
from .qfasm_lexer import QfasmLexer
from .qfasm_utils import *

unaryop = ["sin", "cos", "tan", "exp", "ln", "sqrt", "acos", "atan", "asin"]
unaryop = {"sin": "sin", "cos": "cos", "tan": "tan", "exp": "exp",
"ln": "log", "sqrt": "sqrt", "acos": "arccos", "atan": "arctan", "asin": "arcsin"}
unarynp = {
"sin": np.sin,
"cos": np.cos,
Expand Down Expand Up @@ -84,6 +85,8 @@ def __init__(self, filepath: str = None, debug=False):
self.qnum = 0
# cbit num used
self.cnum = 0
# param
self.params = {}

def add_U_CX(self):
# Add U and CX in global_symtab
Expand Down Expand Up @@ -172,7 +175,9 @@ def handle_gateins(self, gateins: GateInstruction):
for i in range(symnode.num):
tempargs.append(symnode.start + i)
args.append(tempargs)

# change carg to parameter
for i in range(len(gateins.cargs)):
gateins.cargs[i] = self.compute_exp(gateins.cargs[i])
# call many times
for i in range(len(args[0])):
oneargs = []
Expand Down Expand Up @@ -252,52 +257,49 @@ def handle_gateins(self, gateins: GateInstruction):
# change newins's qarg to real q
for i in range(len(newins.qargs)):
newins.qargs[i] = qargdict[newins.qargs[i].name]
# change newins's carg to real carg (consider exp)
# change newins's carg to real carg (consider exp and parameter)
for i in range(len(newins.cargs)):
if not (
isinstance(newins.cargs[i], int)
or isinstance(newins.cargs[i], float)
):
# for expression
newins.cargs[i] = self.compute_exp(newins.cargs[i], cargdict)
# for expression and parameter, it will return parameter or int/float
newins.cargs[i] = self.compute_exp(newins.cargs[i], cargdict)
# now, recurse
gate_list.extend(self.handle_gateins(newins))

return gate_list

def compute_exp(self, carg, cargdict: dict):
def compute_exp(self, carg, cargdict: dict={}):
# recurse
if isinstance(carg, int) or isinstance(carg, float):
if isinstance(carg, int) or isinstance(carg, float) or isinstance(carg, ParameterExpression):
return carg
# if it's id, should get real number from gateins
elif isinstance(carg, Id):
return cargdict[carg.name]
if carg.name in cargdict:
return cargdict[carg.name]
# if it's parameter, just return
else:
return self.params[carg.name]
elif isinstance(carg, UnaryExpr):
if carg.type == "-":
return -self.compute_exp(carg.children[0], cargdict)
elif carg.type in unaryop:
return unarynp[carg.type](self.compute_exp(carg.children[0], cargdict))
nowcarg = self.compute_exp(carg.children[0], cargdict)
if isinstance(nowcarg, ParameterExpression):
func = getattr(nowcarg, unaryop[carg.type])
return func()
else:
return unarynp[carg.type](nowcarg)
elif isinstance(carg, BinaryExpr):
cargl = self.compute_exp(carg.children[0], cargdict)
cargr = self.compute_exp(carg.children[1], cargdict)
if carg.type == "+":
return self.compute_exp(carg.children[0], cargdict) + self.compute_exp(
carg.children[1], cargdict
)
return cargl + cargr
elif carg.type == "-":
return self.compute_exp(carg.children[0], cargdict) - self.compute_exp(
carg.children[1], cargdict
)
return cargl - cargr
elif carg.type == "*":
return self.compute_exp(carg.children[0], cargdict) * self.compute_exp(
carg.children[1], cargdict
)
return cargl * cargr
elif carg.type == "/":
return self.compute_exp(carg.children[0], cargdict) / self.compute_exp(
carg.children[1], cargdict
)
return cargl / cargr
elif carg.type == "^":
return self.compute_exp(carg.children[0], cargdict) ** self.compute_exp(
carg.children[1], cargdict
)
return cargl ** cargr

def addInstruction(self, qc: QuantumCircuit, ins):
if ins is None:
Expand Down Expand Up @@ -416,8 +418,19 @@ def check_qargs(self, gateins: GateInstruction):
f"Qubit used as different argument when call gate {gateins.name} at line {gateins.lineno} file {gateins.filename}"
)

def check_param(self, carg):
if isinstance(carg, int) or isinstance(carg, float):
return
elif isinstance(carg, Id) and carg.name not in self.params:
raise ParserError(f"The parameter {carg.name} is undefined at line {carg.lineno} file {carg.filename}")
elif isinstance(carg, UnaryExpr):
self.check_param(carg.children[0])
elif isinstance(carg, BinaryExpr):
self.check_param(carg.children[0])
self.check_param(carg.children[1])

def check_cargs(self, gateins: GateInstruction):
# check that cargs belongs to unary (they must be int or float)
# check that cargs belongs to unary (they must be int or float or parameter)
# cargs is different from CREG
if gateins.name not in self.nuop and gateins.name not in self.mulctrl:
if gateins.name not in self.global_symtab:
Expand All @@ -429,12 +442,9 @@ def check_cargs(self, gateins: GateInstruction):
raise ParserError(
f"The {gateins.name} is not declared as a gate at line {gateins.lineno} file {gateins.filename}"
)
# check every carg in [int, float]
# check every carg in [int, float, parameter]
for carg in gateins.cargs:
if not (isinstance(carg, int) or isinstance(carg, float)):
raise ParserError(
f"Classical argument must be of type int or float at line {gateins.lineno} file {gateins.filename}"
)
self.check_param(carg)
# check cargs's num matches gate's delcared cargs
if len(gateins.cargs) != len(gatenote.cargs):
raise ParserError(
Expand Down Expand Up @@ -482,11 +492,11 @@ def check_gate_qargs(self, gateins: GateInstruction):

def check_gate_cargs(self, gateins: GateInstruction):
# check gate_op's classcal args, must matches num declared by gate
if gateins.name == "barrier" and len(gateins.cargs) > 0:
if gateins.name in ["barrier", "reset", "measure"] and len(gateins.cargs) > 0:
raise ParserError(
f"Barrier can not receive classical argument at line {gateins.lineno} file {gateins.filename}"
)
if gateins.name != "barrier":
if gateins.name not in ["barrier", "reset", "measure"]:
if gateins.name not in self.global_symtab:
raise ParserError(
f"The gate {gateins.name} is undefined at line {gateins.lineno} file {gateins.filename}"
Expand All @@ -500,7 +510,7 @@ def check_gate_cargs(self, gateins: GateInstruction):
raise ParserError(
f"The number of classical argument declared in gate {gateins.name} is inconsistent with instruction at line {gateins.lineno} file {gateins.filename}"
)
# check carg must from gate declared argument or int/float
# check carg must from gate declared argument or int/float or parameter
for carg in gateins.cargs:
# recurse check expression
self.check_carg_declartion(carg)
Expand All @@ -510,16 +520,17 @@ def check_carg_declartion(self, node):
return
if isinstance(node, Id):
# check declaration
if node.name not in self.symtab:
if node.name in self.symtab:
symnode = self.symtab[node.name]
if symnode.type != "CARG":
raise ParserError(
f"The {node.name} is not declared as a classical bit at line {node.lineno} file {node.filename}"
)
return
elif node.name not in self.params:
raise ParserError(
f"The classical argument {node.name} is undefined at line {node.lineno} file {node.filename}"
)
symnode = self.symtab[node.name]
if symnode.type != "CARG":
raise ParserError(
f"The {node.name} is not declared as a classical bit at line {node.lineno} file {node.filename}"
)
return
if isinstance(node, UnaryExpr):
self.check_carg_declartion(node.children[0])
elif isinstance(node, BinaryExpr):
Expand Down Expand Up @@ -579,7 +590,7 @@ def p_statement_qop(self, p):
| qif error
"""
if p[2] != ";":
raise ParserError(f"Expecting ';' behind statement")
raise ParserError(f"Expecting ';' behind statement at line {p[1].lineno} file {p[1].filename}")
p[0] = p[1]

def p_statement_empty(self, p):
Expand Down Expand Up @@ -979,6 +990,8 @@ def p_statement_bitdecl(self, p):
"""
statement : qdecl ';'
| cdecl ';'
| defparam ';'
| defparam error
| qdecl error
| cdecl error
| error
Expand All @@ -991,6 +1004,20 @@ def p_statement_bitdecl(self, p):
)
p[0] = p[1]

def p_statement_defparam(self, p):
"""
defparam : id EQUAL FLOAT
| id EQUAL INT
| id EQUAL error
"""
if not isinstance(p[3], int) and not isinstance(p[3], float):
raise ParserError(f"Expecting 'INT' or 'FLOAT behind '=' at line {p[1].lineno} file {p[1].filename}")
param_name = p[1].name
if param_name in self.params:
raise ParserError(f"Duplicate declaration for parameter {p[1].name} at line {p[1].lineno} file {p[1].filename}")
self.params[param_name] = Parameter(param_name, p[3])
p[0] = None

def p_qdecl(self, p):
"""
qdecl : QREG indexed_id
Expand Down Expand Up @@ -1169,7 +1196,7 @@ def p_expr_mathfunc(self, p):
"""
if p[1].name not in unaryop:
raise ParserError(
f"Math function {p[1].name} not supported, only support {unaryop} line {p[1].lineno} file {p[1].filename}"
f"Math function {p[1].name} not supported, only support {unaryop.keys()} line {p[1].lineno} file {p[1].filename}"
)
if not isinstance(p[3], Node):
p[0] = unarynp[p[1].name](p[3])
Expand Down
Loading
Loading