forked from M-Labs/artiq
parent
8a28039b74
commit
8d9a22f8da
|
@ -255,6 +255,6 @@ def is_allocated(typ):
|
|||
return not (is_none(typ) or is_bool(typ) or is_int(typ) or
|
||||
is_float(typ) or is_range(typ) or
|
||||
types._is_pointer(typ) or types.is_function(typ) or
|
||||
types.is_c_function(typ) or types.is_rpc_function(typ) or
|
||||
types.is_c_function(typ) or types.is_rpc(typ) or
|
||||
types.is_method(typ) or types.is_tuple(typ) or
|
||||
types.is_value(typ))
|
||||
|
|
|
@ -428,9 +428,8 @@ class StitchingInferencer(Inferencer):
|
|||
if attr_name not in attributes:
|
||||
# We just figured out what the type should be. Add it.
|
||||
attributes[attr_name] = attr_value_type
|
||||
elif not types.is_rpc_function(attr_value_type):
|
||||
else:
|
||||
# Does this conflict with an earlier guess?
|
||||
# RPC function types are exempt because RPCs are dynamically typed.
|
||||
try:
|
||||
attributes[attr_name].unify(attr_value_type)
|
||||
except types.UnificationError as e:
|
||||
|
@ -694,29 +693,22 @@ class Stitcher:
|
|||
# Let the rest of the program decide.
|
||||
return types.TVar()
|
||||
|
||||
def _quote_foreign_function(self, function, loc, syscall, flags):
|
||||
def _quote_syscall(self, function, loc):
|
||||
signature = inspect.signature(function)
|
||||
|
||||
arg_types = OrderedDict()
|
||||
optarg_types = OrderedDict()
|
||||
for param in signature.parameters.values():
|
||||
if param.kind not in (inspect.Parameter.POSITIONAL_ONLY,
|
||||
inspect.Parameter.POSITIONAL_OR_KEYWORD):
|
||||
# We pretend we don't see *args, kwpostargs=..., **kwargs.
|
||||
# Since every method can be still invoked without any arguments
|
||||
# going into *args and the slots after it, this is always safe,
|
||||
# if sometimes constraining.
|
||||
#
|
||||
# Accepting POSITIONAL_ONLY is OK, because the compiler
|
||||
# desugars the keyword arguments into positional ones internally.
|
||||
continue
|
||||
if param.kind != inspect.Parameter.POSITIONAL_OR_KEYWORD:
|
||||
diag = diagnostic.Diagnostic("error",
|
||||
"system calls must only use positional arguments; '{argument}' isn't",
|
||||
{"argument": param.name},
|
||||
self._function_loc(function),
|
||||
notes=self._call_site_note(loc, is_syscall=True))
|
||||
self.engine.process(diag)
|
||||
|
||||
if param.default is inspect.Parameter.empty:
|
||||
arg_types[param.name] = self._type_of_param(function, loc, param,
|
||||
is_syscall=syscall is not None)
|
||||
elif syscall is None:
|
||||
optarg_types[param.name] = self._type_of_param(function, loc, param,
|
||||
is_syscall=False)
|
||||
arg_types[param.name] = self._type_of_param(function, loc, param, is_syscall=True)
|
||||
else:
|
||||
diag = diagnostic.Diagnostic("error",
|
||||
"system call argument '{argument}' must not have a default value",
|
||||
|
@ -727,10 +719,8 @@ class Stitcher:
|
|||
|
||||
if signature.return_annotation is not inspect.Signature.empty:
|
||||
ret_type = self._extract_annot(function, signature.return_annotation,
|
||||
"return type", loc, is_syscall=syscall is not None)
|
||||
elif syscall is None:
|
||||
ret_type = builtins.TNone()
|
||||
else: # syscall is not None
|
||||
"return type", loc, is_syscall=True)
|
||||
else:
|
||||
diag = diagnostic.Diagnostic("error",
|
||||
"system call must have a return type annotation", {},
|
||||
self._function_loc(function),
|
||||
|
@ -738,15 +728,23 @@ class Stitcher:
|
|||
self.engine.process(diag)
|
||||
ret_type = types.TVar()
|
||||
|
||||
if syscall is None:
|
||||
function_type = types.TRPCFunction(arg_types, optarg_types, ret_type,
|
||||
service=self.object_map.store(function))
|
||||
else:
|
||||
function_type = types.TCFunction(arg_types, ret_type,
|
||||
name=syscall, flags=flags)
|
||||
|
||||
name=function.artiq_embedded.syscall,
|
||||
flags=function.artiq_embedded.flags)
|
||||
self.functions[function] = function_type
|
||||
return function_type
|
||||
|
||||
def _quote_rpc(self, function, loc):
|
||||
signature = inspect.signature(function)
|
||||
|
||||
if signature.return_annotation is not inspect.Signature.empty:
|
||||
ret_type = self._extract_annot(function, signature.return_annotation,
|
||||
"return type", loc, is_syscall=False)
|
||||
else:
|
||||
ret_type = builtins.TNone()
|
||||
|
||||
function_type = types.TRPC(ret_type, service=self.object_map.store(function))
|
||||
self.functions[function] = function_type
|
||||
return function_type
|
||||
|
||||
def _quote_function(self, function, loc):
|
||||
|
@ -780,9 +778,7 @@ class Stitcher:
|
|||
elif function.artiq_embedded.syscall is not None:
|
||||
# Insert a storage-less global whose type instructs the compiler
|
||||
# to perform a system call instead of a regular call.
|
||||
self._quote_foreign_function(function, loc,
|
||||
syscall=function.artiq_embedded.syscall,
|
||||
flags=function.artiq_embedded.flags)
|
||||
self._quote_syscall(function, loc)
|
||||
elif function.artiq_embedded.forbidden is not None:
|
||||
diag = diagnostic.Diagnostic("fatal",
|
||||
"this function cannot be called as an RPC", {},
|
||||
|
@ -792,14 +788,9 @@ class Stitcher:
|
|||
else:
|
||||
assert False
|
||||
else:
|
||||
# Insert a storage-less global whose type instructs the compiler
|
||||
# to perform an RPC instead of a regular call.
|
||||
self._quote_foreign_function(function, loc, syscall=None, flags=None)
|
||||
self._quote_rpc(function, loc)
|
||||
|
||||
function_type = self.functions[function]
|
||||
if types.is_rpc_function(function_type):
|
||||
function_type = types.instantiate(function_type)
|
||||
return function_type
|
||||
return self.functions[function]
|
||||
|
||||
def _quote(self, value, loc):
|
||||
synthesizer = self._synthesizer(loc)
|
||||
|
|
|
@ -23,12 +23,19 @@ def is_basic_block(typ):
|
|||
return isinstance(typ, TBasicBlock)
|
||||
|
||||
class TOption(types.TMono):
|
||||
def __init__(self, inner):
|
||||
super().__init__("option", {"inner": inner})
|
||||
def __init__(self, value):
|
||||
super().__init__("option", {"value": value})
|
||||
|
||||
def is_option(typ):
|
||||
return isinstance(typ, TOption)
|
||||
|
||||
class TKeyword(types.TMono):
|
||||
def __init__(self, value):
|
||||
super().__init__("keyword", {"value": value})
|
||||
|
||||
def is_keyword(typ):
|
||||
return isinstance(typ, TKeyword)
|
||||
|
||||
class TExceptionTypeInfo(types.TMono):
|
||||
def __init__(self):
|
||||
super().__init__("exntypeinfo")
|
||||
|
@ -678,7 +685,7 @@ class GetAttr(Instruction):
|
|||
typ = obj.type.attributes[attr]
|
||||
else:
|
||||
typ = obj.type.constructor.attributes[attr]
|
||||
if types.is_function(typ):
|
||||
if types.is_function(typ) or types.is_rpc(typ):
|
||||
typ = types.TMethod(obj.type, typ)
|
||||
super().__init__([obj], typ, name)
|
||||
self.attr = attr
|
||||
|
|
|
@ -922,9 +922,6 @@ class ARTIQIRGenerator(algorithm.Visitor):
|
|||
if self.current_assign is None:
|
||||
return self.append(ir.GetAttr(obj, node.attr,
|
||||
name="{}.FLD.{}".format(_readable_name(obj), node.attr)))
|
||||
elif types.is_rpc_function(self.current_assign.type):
|
||||
# RPC functions are just type-level markers
|
||||
return self.append(ir.Builtin("nop", [], builtins.TNone()))
|
||||
else:
|
||||
return self.append(ir.SetAttr(obj, node.attr, self.current_assign))
|
||||
|
||||
|
@ -1719,7 +1716,7 @@ class ARTIQIRGenerator(algorithm.Visitor):
|
|||
self.engine.process(diag)
|
||||
|
||||
def _user_call(self, callee, positional, keywords, arg_exprs={}):
|
||||
if types.is_function(callee.type):
|
||||
if types.is_function(callee.type) or types.is_rpc(callee.type):
|
||||
func = callee
|
||||
self_arg = None
|
||||
fn_typ = callee.type
|
||||
|
@ -1734,6 +1731,17 @@ class ARTIQIRGenerator(algorithm.Visitor):
|
|||
else:
|
||||
assert False
|
||||
|
||||
if types.is_rpc(fn_typ):
|
||||
if self_arg is None:
|
||||
args = positional
|
||||
else:
|
||||
args = [self_arg] + positional
|
||||
|
||||
for keyword in keywords:
|
||||
arg = keywords[keyword]
|
||||
args.append(self.append(ir.Alloc([ir.Constant(keyword, builtins.TStr()), arg],
|
||||
ir.TKeyword(arg.type))))
|
||||
else:
|
||||
args = [None] * (len(fn_typ.args) + len(fn_typ.optargs))
|
||||
|
||||
for index, arg in enumerate(positional):
|
||||
|
|
|
@ -109,17 +109,11 @@ class Inferencer(algorithm.Visitor):
|
|||
]
|
||||
|
||||
attr_type = object_type.attributes[attr_name]
|
||||
if types.is_rpc_function(attr_type):
|
||||
attr_type = types.instantiate(attr_type)
|
||||
|
||||
self._unify(result_type, attr_type, loc, None,
|
||||
makenotes=makenotes, when=" for attribute '{}'".format(attr_name))
|
||||
elif types.is_instance(object_type) and \
|
||||
attr_name in object_type.constructor.attributes:
|
||||
attr_type = object_type.constructor.attributes[attr_name].find()
|
||||
if types.is_rpc_function(attr_type):
|
||||
attr_type = types.instantiate(attr_type)
|
||||
|
||||
if types.is_function(attr_type):
|
||||
# Convert to a method.
|
||||
if len(attr_type.args) < 1:
|
||||
|
@ -155,6 +149,10 @@ class Inferencer(algorithm.Visitor):
|
|||
when=" while inferring the type for self argument")
|
||||
|
||||
attr_type = types.TMethod(object_type, attr_type)
|
||||
elif types.is_rpc(attr_type):
|
||||
# Convert to a method. We don't have to bother typechecking
|
||||
# the self argument, since for RPCs anything goes.
|
||||
attr_type = types.TMethod(object_type, attr_type)
|
||||
|
||||
if not types.is_var(attr_type):
|
||||
self._unify(result_type, attr_type,
|
||||
|
@ -871,6 +869,10 @@ class Inferencer(algorithm.Visitor):
|
|||
return # not enough info yet
|
||||
elif types.is_builtin(typ):
|
||||
return self.visit_builtin_call(node)
|
||||
elif types.is_rpc(typ):
|
||||
self._unify(node.type, typ.ret,
|
||||
node.loc, None)
|
||||
return
|
||||
elif not (types.is_function(typ) or types.is_method(typ)):
|
||||
diag = diagnostic.Diagnostic("error",
|
||||
"cannot call this expression of type {type}",
|
||||
|
@ -888,6 +890,10 @@ class Inferencer(algorithm.Visitor):
|
|||
typ = types.get_method_function(typ)
|
||||
if types.is_var(typ):
|
||||
return # not enough info yet
|
||||
elif types.is_rpc(typ):
|
||||
self._unify(node.type, typ.ret,
|
||||
node.loc, None)
|
||||
return
|
||||
|
||||
typ_arity = typ.arity() - 1
|
||||
typ_args = OrderedDict(list(typ.args.items())[1:])
|
||||
|
|
|
@ -280,7 +280,7 @@ class IODelayEstimator(algorithm.Visitor):
|
|||
context="as an argument for delay_mu()")
|
||||
call_delay = value
|
||||
elif not types.is_builtin(typ):
|
||||
if types.is_function(typ):
|
||||
if types.is_function(typ) or types.is_rpc(typ):
|
||||
offset = 0
|
||||
elif types.is_method(typ):
|
||||
offset = 1
|
||||
|
@ -288,6 +288,9 @@ class IODelayEstimator(algorithm.Visitor):
|
|||
else:
|
||||
assert False
|
||||
|
||||
if types.is_rpc(typ):
|
||||
call_delay = iodelay.Const(0)
|
||||
else:
|
||||
delay = typ.find().delay.find()
|
||||
if types.is_var(delay):
|
||||
raise _UnknownDelay()
|
||||
|
|
|
@ -177,7 +177,7 @@ class LLVMIRGenerator:
|
|||
typ = typ.find()
|
||||
if types.is_tuple(typ):
|
||||
return ll.LiteralStructType([self.llty_of_type(eltty) for eltty in typ.elts])
|
||||
elif types.is_rpc_function(typ) or types.is_c_function(typ):
|
||||
elif types.is_rpc(typ) or types.is_c_function(typ):
|
||||
if for_return:
|
||||
return llvoid
|
||||
else:
|
||||
|
@ -229,7 +229,9 @@ class LLVMIRGenerator:
|
|||
elif ir.is_basic_block(typ):
|
||||
return llptr
|
||||
elif ir.is_option(typ):
|
||||
return ll.LiteralStructType([lli1, self.llty_of_type(typ.params["inner"])])
|
||||
return ll.LiteralStructType([lli1, self.llty_of_type(typ.params["value"])])
|
||||
elif ir.is_keyword(typ):
|
||||
return ll.LiteralStructType([llptr, self.llty_of_type(typ.params["value"])])
|
||||
elif ir.is_environment(typ):
|
||||
llty = self.llcontext.get_identified_type("env.{}".format(typ.env_name))
|
||||
if llty.elements is None:
|
||||
|
@ -618,7 +620,7 @@ class LLVMIRGenerator:
|
|||
size=llsize)
|
||||
llvalue = self.llbuilder.insert_value(llvalue, llalloc, 1, name=insn.name)
|
||||
return llvalue
|
||||
elif not builtins.is_allocated(insn.type):
|
||||
elif not builtins.is_allocated(insn.type) or ir.is_keyword(insn.type):
|
||||
llvalue = ll.Constant(self.llty_of_type(insn.type), ll.Undefined)
|
||||
for index, elt in enumerate(insn.operands):
|
||||
llvalue = self.llbuilder.insert_value(llvalue, self.map(elt), index)
|
||||
|
@ -707,8 +709,8 @@ class LLVMIRGenerator:
|
|||
def get_global_closure(self, typ, attr):
|
||||
closure_type = typ.attributes[attr]
|
||||
assert types.is_constructor(typ)
|
||||
assert types.is_function(closure_type)
|
||||
if types.is_c_function(closure_type) or types.is_rpc_function(closure_type):
|
||||
assert types.is_function(closure_type) or types.is_rpc(closure_type)
|
||||
if types.is_c_function(closure_type) or types.is_rpc(closure_type):
|
||||
return None
|
||||
|
||||
llty = self.llty_of_type(typ.attributes[attr])
|
||||
|
@ -1156,8 +1158,8 @@ class LLVMIRGenerator:
|
|||
elif builtins.is_range(typ):
|
||||
return b"r" + self._rpc_tag(builtins.get_iterable_elt(typ),
|
||||
error_handler)
|
||||
elif ir.is_option(typ):
|
||||
return b"o" + self._rpc_tag(typ.params["inner"],
|
||||
elif ir.is_keyword(typ):
|
||||
return b"k" + self._rpc_tag(typ.params["value"],
|
||||
error_handler)
|
||||
elif '__objectid__' in typ.attributes:
|
||||
return b"O"
|
||||
|
@ -1271,7 +1273,7 @@ class LLVMIRGenerator:
|
|||
|
||||
def process_Call(self, insn):
|
||||
functiontyp = insn.target_function().type
|
||||
if types.is_rpc_function(functiontyp):
|
||||
if types.is_rpc(functiontyp):
|
||||
return self._build_rpc(insn.target_function().loc,
|
||||
functiontyp,
|
||||
insn.arguments(),
|
||||
|
@ -1303,7 +1305,7 @@ class LLVMIRGenerator:
|
|||
functiontyp = insn.target_function().type
|
||||
llnormalblock = self.map(insn.normal_target())
|
||||
llunwindblock = self.map(insn.exception_target())
|
||||
if types.is_rpc_function(functiontyp):
|
||||
if types.is_rpc(functiontyp):
|
||||
return self._build_rpc(insn.target_function().loc,
|
||||
functiontyp,
|
||||
insn.arguments(),
|
||||
|
@ -1392,7 +1394,7 @@ class LLVMIRGenerator:
|
|||
lleltsptr = llglobal.bitcast(lleltsary.type.element.as_pointer())
|
||||
llconst = ll.Constant(llty, [ll.Constant(lli32, len(llelts)), lleltsptr])
|
||||
return llconst
|
||||
elif types.is_function(typ):
|
||||
elif types.is_rpc(typ) or types.is_function(typ):
|
||||
# RPC and C functions have no runtime representation.
|
||||
# We only get down this codepath for ARTIQ Python functions when they're
|
||||
# referenced from a constructor, and the value inside the constructor
|
||||
|
|
|
@ -94,9 +94,6 @@ class TVar(Type):
|
|||
else:
|
||||
return self.find().fold(accum, fn)
|
||||
|
||||
def map(self, fn):
|
||||
return fn(self)
|
||||
|
||||
def __repr__(self):
|
||||
if self.parent is self:
|
||||
return "<artiq.compiler.types.TVar %d>" % id(self)
|
||||
|
@ -141,21 +138,6 @@ class TMono(Type):
|
|||
accum = self.params[param].fold(accum, fn)
|
||||
return fn(accum, self)
|
||||
|
||||
def map(self, fn):
|
||||
params = OrderedDict()
|
||||
for param in self.params:
|
||||
params[param] = self.params[param].map(fn)
|
||||
|
||||
attributes = OrderedDict()
|
||||
for attr in self.attributes:
|
||||
attributes[attr] = self.attributes[attr].map(fn)
|
||||
|
||||
self_copy = self.__class__.__new__(self.__class__)
|
||||
self_copy.name = self.name
|
||||
self_copy.params = params
|
||||
self_copy.attributes = attributes
|
||||
return fn(self_copy)
|
||||
|
||||
def __repr__(self):
|
||||
return "artiq.compiler.types.TMono(%s, %s)" % (repr(self.name), repr(self.params))
|
||||
|
||||
|
@ -202,9 +184,6 @@ class TTuple(Type):
|
|||
accum = elt.fold(accum, fn)
|
||||
return fn(accum, self)
|
||||
|
||||
def map(self, fn):
|
||||
return fn(TTuple(list(map(lambda elt: elt.map(fn), self.elts))))
|
||||
|
||||
def __repr__(self):
|
||||
return "artiq.compiler.types.TTuple(%s)" % repr(self.elts)
|
||||
|
||||
|
@ -276,23 +255,6 @@ class TFunction(Type):
|
|||
accum = self.ret.fold(accum, fn)
|
||||
return fn(accum, self)
|
||||
|
||||
def _map_args(self, fn):
|
||||
args = OrderedDict()
|
||||
for arg in self.args:
|
||||
args[arg] = self.args[arg].map(fn)
|
||||
|
||||
optargs = OrderedDict()
|
||||
for optarg in self.optargs:
|
||||
optargs[optarg] = self.optargs[optarg].map(fn)
|
||||
|
||||
return args, optargs, self.ret.map(fn)
|
||||
|
||||
def map(self, fn):
|
||||
args, optargs, ret = self._map_args(fn)
|
||||
self_copy = TFunction(args, optargs, ret)
|
||||
self_copy.delay = self.delay.map(fn)
|
||||
return fn(self_copy)
|
||||
|
||||
def __repr__(self):
|
||||
return "artiq.compiler.types.TFunction({}, {}, {})".format(
|
||||
repr(self.args), repr(self.optargs), repr(self.ret))
|
||||
|
@ -308,35 +270,6 @@ class TFunction(Type):
|
|||
def __hash__(self):
|
||||
return hash((_freeze(self.args), _freeze(self.optargs), self.ret))
|
||||
|
||||
class TRPCFunction(TFunction):
|
||||
"""
|
||||
A function type of a remote function.
|
||||
|
||||
:ivar service: (int) RPC service number
|
||||
"""
|
||||
|
||||
attributes = OrderedDict()
|
||||
|
||||
def __init__(self, args, optargs, ret, service):
|
||||
super().__init__(args, optargs, ret)
|
||||
self.service = service
|
||||
self.delay = TFixedDelay(iodelay.Const(0))
|
||||
|
||||
def unify(self, other):
|
||||
if isinstance(other, TRPCFunction) and \
|
||||
self.service == other.service:
|
||||
super().unify(other)
|
||||
elif isinstance(other, TVar):
|
||||
other.unify(self)
|
||||
else:
|
||||
raise UnificationError(self, other)
|
||||
|
||||
def map(self, fn):
|
||||
args, optargs, ret = self._map_args(fn)
|
||||
self_copy = TRPCFunction(args, optargs, ret, self.service)
|
||||
self_copy.delay = self.delay.map(fn)
|
||||
return fn(self_copy)
|
||||
|
||||
class TCFunction(TFunction):
|
||||
"""
|
||||
A function type of a runtime-provided C function.
|
||||
|
@ -368,11 +301,49 @@ class TCFunction(TFunction):
|
|||
else:
|
||||
raise UnificationError(self, other)
|
||||
|
||||
def map(self, fn):
|
||||
args, _optargs, ret = self._map_args(fn)
|
||||
self_copy = TCFunction(args, ret, self.name)
|
||||
self_copy.delay = self.delay.map(fn)
|
||||
return fn(self_copy)
|
||||
class TRPC(Type):
|
||||
"""
|
||||
A type of a remote call.
|
||||
|
||||
:ivar ret: (:class:`Type`)
|
||||
return type
|
||||
:ivar service: (int) RPC service number
|
||||
"""
|
||||
|
||||
attributes = OrderedDict()
|
||||
|
||||
def __init__(self, ret, service):
|
||||
assert isinstance(ret, Type)
|
||||
self.ret, self.service = ret, service
|
||||
|
||||
def find(self):
|
||||
return self
|
||||
|
||||
def unify(self, other):
|
||||
if isinstance(other, TRPC) and \
|
||||
self.service == other.service:
|
||||
self.ret.unify(other.ret)
|
||||
elif isinstance(other, TVar):
|
||||
other.unify(self)
|
||||
else:
|
||||
raise UnificationError(self, other)
|
||||
|
||||
def fold(self, accum, fn):
|
||||
accum = self.ret.fold(accum, fn)
|
||||
return fn(accum, self)
|
||||
|
||||
def __repr__(self):
|
||||
return "artiq.compiler.types.TRPC({})".format(repr(self.ret))
|
||||
|
||||
def __eq__(self, other):
|
||||
return isinstance(other, TRPC) and \
|
||||
self.service == other.service
|
||||
|
||||
def __ne__(self, other):
|
||||
return not (self == other)
|
||||
|
||||
def __hash__(self):
|
||||
return hash(self.service)
|
||||
|
||||
class TBuiltin(Type):
|
||||
"""
|
||||
|
@ -395,9 +366,6 @@ class TBuiltin(Type):
|
|||
def fold(self, accum, fn):
|
||||
return fn(accum, self)
|
||||
|
||||
def map(self, fn):
|
||||
return fn(self)
|
||||
|
||||
def __repr__(self):
|
||||
return "artiq.compiler.types.{}({})".format(type(self).__name__, repr(self.name))
|
||||
|
||||
|
@ -490,9 +458,6 @@ class TValue(Type):
|
|||
def fold(self, accum, fn):
|
||||
return fn(accum, self)
|
||||
|
||||
def map(self, fn):
|
||||
return fn(self)
|
||||
|
||||
def __repr__(self):
|
||||
return "artiq.compiler.types.TValue(%s)" % repr(self.value)
|
||||
|
||||
|
@ -543,10 +508,6 @@ class TDelay(Type):
|
|||
# delay types do not participate in folding
|
||||
pass
|
||||
|
||||
def map(self, fn):
|
||||
# or mapping
|
||||
return self
|
||||
|
||||
def __eq__(self, other):
|
||||
return isinstance(other, TDelay) and \
|
||||
(self.duration == other.duration and \
|
||||
|
@ -570,18 +531,6 @@ def TFixedDelay(duration):
|
|||
return TDelay(duration, None)
|
||||
|
||||
|
||||
def instantiate(typ):
|
||||
tvar_map = dict()
|
||||
def mapper(typ):
|
||||
typ = typ.find()
|
||||
if is_var(typ):
|
||||
if typ not in tvar_map:
|
||||
tvar_map[typ] = TVar()
|
||||
return tvar_map[typ]
|
||||
return typ
|
||||
|
||||
return typ.map(mapper)
|
||||
|
||||
def is_var(typ):
|
||||
return isinstance(typ.find(), TVar)
|
||||
|
||||
|
@ -616,8 +565,8 @@ def _is_pointer(typ):
|
|||
def is_function(typ):
|
||||
return isinstance(typ.find(), TFunction)
|
||||
|
||||
def is_rpc_function(typ):
|
||||
return isinstance(typ.find(), TRPCFunction)
|
||||
def is_rpc(typ):
|
||||
return isinstance(typ.find(), TRPC)
|
||||
|
||||
def is_c_function(typ, name=None):
|
||||
typ = typ.find()
|
||||
|
@ -732,7 +681,7 @@ class TypePrinter(object):
|
|||
return "(%s,)" % self.name(typ.elts[0], depth + 1)
|
||||
else:
|
||||
return "(%s)" % ", ".join([self.name(typ, depth + 1) for typ in typ.elts])
|
||||
elif isinstance(typ, (TFunction, TRPCFunction, TCFunction)):
|
||||
elif isinstance(typ, (TFunction, TCFunction)):
|
||||
args = []
|
||||
args += [ "%s:%s" % (arg, self.name(typ.args[arg], depth + 1))
|
||||
for arg in typ.args]
|
||||
|
@ -746,12 +695,12 @@ class TypePrinter(object):
|
|||
elif not (delay.is_fixed() and iodelay.is_zero(delay.duration)):
|
||||
signature += " " + self.name(delay, depth + 1)
|
||||
|
||||
if isinstance(typ, TRPCFunction):
|
||||
return "[rpc #{}]{}".format(typ.service, signature)
|
||||
if isinstance(typ, TCFunction):
|
||||
return "[ffi {}]{}".format(repr(typ.name), signature)
|
||||
elif isinstance(typ, TFunction):
|
||||
return signature
|
||||
elif isinstance(typ, TRPC):
|
||||
return "[rpc #{}](...)->{}".format(typ.service, self.name(typ.ret, depth + 1))
|
||||
elif isinstance(typ, TBuiltinFunction):
|
||||
return "<function {}>".format(typ.name)
|
||||
elif isinstance(typ, (TConstructor, TExceptionConstructor)):
|
||||
|
|
|
@ -30,7 +30,7 @@ class MonomorphismValidator(algorithm.Visitor):
|
|||
super().generic_visit(node)
|
||||
|
||||
if isinstance(node, asttyped.commontyped):
|
||||
if types.is_polymorphic(node.type) and not types.is_rpc_function(node.type):
|
||||
if types.is_polymorphic(node.type):
|
||||
note = diagnostic.Diagnostic("note",
|
||||
"the expression has type {type}",
|
||||
{"type": types.TypePrinter().name(node.type)},
|
||||
|
|
|
@ -3,6 +3,7 @@ import logging
|
|||
import traceback
|
||||
from enum import Enum
|
||||
from fractions import Fraction
|
||||
from collections import namedtuple
|
||||
|
||||
from artiq.coredevice import exceptions
|
||||
from artiq.language.core import int as wrapping_int
|
||||
|
@ -62,6 +63,9 @@ class RPCReturnValueError(ValueError):
|
|||
pass
|
||||
|
||||
|
||||
RPCKeyword = namedtuple('RPCKeyword', ['name', 'value'])
|
||||
|
||||
|
||||
class CommGeneric:
|
||||
def __init__(self):
|
||||
self._read_type = self._write_type = None
|
||||
|
@ -229,7 +233,8 @@ class CommGeneric:
|
|||
raise UnsupportedDevice("Unsupported runtime ID: {}"
|
||||
.format(runtime_id))
|
||||
gateware_version = self._read_chunk(self._read_length).decode("utf-8")
|
||||
if gateware_version != software_version:
|
||||
if gateware_version != software_version and \
|
||||
gateware_version + ".dirty" != software_version:
|
||||
logger.warning("Mismatch between gateware (%s) "
|
||||
"and software (%s) versions",
|
||||
gateware_version, software_version)
|
||||
|
@ -298,7 +303,6 @@ class CommGeneric:
|
|||
logger.debug("running kernel")
|
||||
|
||||
_rpc_sentinel = object()
|
||||
_rpc_undefined = object()
|
||||
|
||||
# See session.c:{send,receive}_rpc_value and llvm_ir_generator.py:_rpc_tag.
|
||||
def _receive_rpc_value(self, object_map):
|
||||
|
@ -332,27 +336,23 @@ class CommGeneric:
|
|||
stop = self._receive_rpc_value(object_map)
|
||||
step = self._receive_rpc_value(object_map)
|
||||
return range(start, stop, step)
|
||||
elif tag == "o":
|
||||
present = self._read_int8()
|
||||
if present:
|
||||
return self._receive_rpc_value(object_map)
|
||||
else:
|
||||
return self._rpc_undefined
|
||||
elif tag == "k":
|
||||
name = self._read_string()
|
||||
value = self._receive_rpc_value(object_map)
|
||||
return RPCKeyword(name, value)
|
||||
elif tag == "O":
|
||||
return object_map.retrieve(self._read_int32())
|
||||
else:
|
||||
raise IOError("Unknown RPC value tag: {}".format(repr(tag)))
|
||||
|
||||
def _receive_rpc_args(self, object_map, defaults):
|
||||
args = []
|
||||
default_arg_num = 0
|
||||
args, kwargs = [], {}
|
||||
while True:
|
||||
value = self._receive_rpc_value(object_map)
|
||||
if value is self._rpc_sentinel:
|
||||
return args
|
||||
elif value is self._rpc_undefined:
|
||||
args.append(defaults[default_arg_num])
|
||||
default_arg_num += 1
|
||||
return args, kwargs
|
||||
elif isinstance(value, RPCKeyword):
|
||||
kwargs[value.name] = value.value
|
||||
else:
|
||||
args.append(value)
|
||||
|
||||
|
@ -443,13 +443,13 @@ class CommGeneric:
|
|||
else:
|
||||
service = object_map.retrieve(service_id)
|
||||
|
||||
arguments = self._receive_rpc_args(object_map, service.__defaults__)
|
||||
args, kwargs = self._receive_rpc_args(object_map, service.__defaults__)
|
||||
return_tags = self._read_bytes()
|
||||
logger.debug("rpc service: [%d]%r %r -> %s", service_id, service, arguments, return_tags)
|
||||
logger.debug("rpc service: [%d]%r %r %r -> %s", service_id, service, args, kwargs, return_tags)
|
||||
|
||||
try:
|
||||
result = service(*arguments)
|
||||
logger.debug("rpc service: %d %r == %r", service_id, arguments, result)
|
||||
result = service(*args, **kwargs)
|
||||
logger.debug("rpc service: %d %r %r == %r", service_id, args, kwargs, result)
|
||||
|
||||
if service_id != 0:
|
||||
self._write_header(_H2DMsgType.RPC_REPLY)
|
||||
|
@ -457,7 +457,7 @@ class CommGeneric:
|
|||
self._send_rpc_value(bytearray(return_tags), result, result, service)
|
||||
self._write_flush()
|
||||
except Exception as exn:
|
||||
logger.debug("rpc service: %d %r ! %r", service_id, arguments, exn)
|
||||
logger.debug("rpc service: %d %r %r ! %r", service_id, args, kwargs, exn)
|
||||
|
||||
self._write_header(_H2DMsgType.RPC_EXCEPTION)
|
||||
|
||||
|
@ -486,7 +486,13 @@ class CommGeneric:
|
|||
for index in range(3):
|
||||
self._write_int64(0)
|
||||
|
||||
(_, (filename, line, function, _), ) = traceback.extract_tb(exn.__traceback__, 2)
|
||||
tb = traceback.extract_tb(exn.__traceback__, 2)
|
||||
if len(tb) == 2:
|
||||
(_, (filename, line, function, _), ) = tb
|
||||
elif len(tb) == 1:
|
||||
((filename, line, function, _), ) = tb
|
||||
else:
|
||||
assert False
|
||||
self._write_string(filename)
|
||||
self._write_int32(line)
|
||||
self._write_int32(-1) # column not known
|
||||
|
|
|
@ -862,21 +862,16 @@ static int send_rpc_value(const char **tag, void **value)
|
|||
break;
|
||||
}
|
||||
|
||||
case 'o': { // option(inner='a)
|
||||
struct { int8_t present; struct {} contents; } *option = *value;
|
||||
case 'k': { // keyword(value='a)
|
||||
struct { const char *name; struct {} contents; } *option = *value;
|
||||
void *contents = &option->contents;
|
||||
|
||||
if(!out_packet_int8(option->present))
|
||||
if(!out_packet_string(option->name))
|
||||
return 0;
|
||||
|
||||
// option never appears in composite types, so we don't have
|
||||
// keyword never appears in composite types, so we don't have
|
||||
// to accurately advance *value.
|
||||
if(option->present) {
|
||||
return send_rpc_value(tag, &contents);
|
||||
} else {
|
||||
skip_rpc_value(tag);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
case 'O': { // host object
|
||||
|
|
|
@ -59,6 +59,56 @@ class DefaultArgTest(ExperimentCase):
|
|||
self.assertEqual(exp.run(), 42)
|
||||
|
||||
|
||||
class _RPC(EnvExperiment):
|
||||
def build(self):
|
||||
self.setattr_device("core")
|
||||
|
||||
def args(self, *args) -> TInt32:
|
||||
return len(args)
|
||||
|
||||
def kwargs(self, x="", **kwargs) -> TInt32:
|
||||
return len(kwargs)
|
||||
|
||||
@kernel
|
||||
def args0(self):
|
||||
return self.args()
|
||||
|
||||
@kernel
|
||||
def args1(self):
|
||||
return self.args("A")
|
||||
|
||||
@kernel
|
||||
def args2(self):
|
||||
return self.args("A", 1)
|
||||
|
||||
@kernel
|
||||
def kwargs0(self):
|
||||
return self.kwargs()
|
||||
|
||||
@kernel
|
||||
def kwargs1(self):
|
||||
return self.kwargs(a="A")
|
||||
|
||||
@kernel
|
||||
def kwargs2(self):
|
||||
return self.kwargs(a="A", b=1)
|
||||
|
||||
@kernel
|
||||
def args1kwargs2(self):
|
||||
return self.kwargs("X", a="A", b=1)
|
||||
|
||||
class RPCTest(ExperimentCase):
|
||||
def test_args(self):
|
||||
exp = self.create(_RPC)
|
||||
self.assertEqual(exp.args0(), 0)
|
||||
self.assertEqual(exp.args1(), 1)
|
||||
self.assertEqual(exp.args2(), 2)
|
||||
self.assertEqual(exp.kwargs0(), 0)
|
||||
self.assertEqual(exp.kwargs1(), 1)
|
||||
self.assertEqual(exp.kwargs2(), 2)
|
||||
self.assertEqual(exp.args1kwargs2(), 2)
|
||||
|
||||
|
||||
class _Payload1MB(EnvExperiment):
|
||||
def build(self):
|
||||
self.setattr_device("core")
|
||||
|
|
|
@ -1,15 +0,0 @@
|
|||
# RUN: %python -m artiq.compiler.testbench.embedding +diag %s >%t
|
||||
# RUN: OutputCheck %s --file-to-check=%t
|
||||
|
||||
from artiq.language.core import *
|
||||
from artiq.language.types import *
|
||||
|
||||
# CHECK-L: <synthesized>:1: error: cannot unify int(width='a) with str
|
||||
# CHECK-L: ${LINE:+1}: note: expanded from here while trying to infer a type for an unannotated optional argument 'x' from its default value
|
||||
def foo(x=[1,"x"]):
|
||||
pass
|
||||
|
||||
@kernel
|
||||
def entrypoint():
|
||||
# CHECK-L: ${LINE:+1}: note: in function called remotely here
|
||||
foo()
|
Loading…
Reference in New Issue