occheung
/
artiq
forked from M-Labs/artiq
1
0
Fork 0
Code Pull Requests Activity

compiler: Explain use of rpc_tag() in array ops, formatting [nfc]

This commit is contained in:
David Nadlinger 2020-07-29 23:19:01 +01:00
parent a7e855b319
commit 4d002c7934
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
artiq/compiler/transforms/artiq_ir_generator.py
View File

@ -1429,13 +1429,17 @@ class ARTIQIRGenerator(algorithm.Visitor):
# differ. # differ.
def name_error(typ): def name_error(typ):
assert False, "Internal compiler error: No RPC tag for {}".format(typ) assert False, "Internal compiler error: No RPC tag for {}".format(typ)
def mangle_name(typ): def mangle_name(typ):
typ = typ.find() typ = typ.find()
return ir.rpc_tag(typ["elt"], name_error).decode() +\ # rpc_tag is used to turn element types into mangled names for no
str(typ["num_dims"].find().value) # particularly good reason apart from not having to invent yet another
# string representation.
return (ir.rpc_tag(typ["elt"], name_error).decode() +
str(typ["num_dims"].find().value))
name = "_array_{}_{}_{}_{}".format( name = "_array_{}_{}_{}_{}".format(
type(op).__name__, type(op).__name__, *(map(mangle_name, (result_type, lhs_type, rhs_type))))
*(map(mangle_name, (result_type, lhs_type, rhs_type))))
if name not in self.array_binop_funcs: if name not in self.array_binop_funcs:
self.array_binop_funcs[name] = self._make_array_binop( self.array_binop_funcs[name] = self._make_array_binop(
name, op, result_type, lhs_type, rhs_type) name, op, result_type, lhs_type, rhs_type)