def generic(self, args, kws):
assert not kws
assert len(args) == 4
if args[2] == types.intp: # string read call, returns string array
return signature(string_array_type, *unliteral_all(args))
# array_ty = types.Array(ndim=1, layout='C', dtype=args[2])
return signature(types.int64, *unliteral_all(args))
def generic(self, args, kws):
if len(args) == 1 and isinstance(args[0], DictKeyIteratorType):
return signature(args[0].key_typ, *unliteral_all(args))
def resolve_pop(self, dict, args, kws):
assert not kws
return signature(dict.val_typ, *unliteral_all(args))
def resolve_get(self, dict, args, kws):
assert not kws
assert len(args) == 2
return signature(args[1], *unliteral_all(args))
def generic(self, args, kws):
assert not kws
assert len(args) == 6
# array_ty = types.Array(ndim=1, layout='C', dtype=args[2])
return signature(types.int32, *unliteral_all(args))
def generic(self, args, kws):
assert not kws
assert len(args) == 4
return signature(string_array_type, *unliteral_all(args))
def generic(self, args, kws):
assert not kws
[ary, idx, val] = args
if isinstance(ary, ReqArrayType
) and idx == types.intp and val == mpi_req_numba_type:
return signature(types.none, *unliteral_all(args))
def generic(self, args, kws):
assert not kws
assert len(args) == 2 and args == (types.int32, req_array_type)
return signature(types.none, *unliteral_all(args))
def generic(self, args, kws):
assert not kws
assert len(args) == 1 and args[0] == req_array_type
return signature(types.none, *unliteral_all(args))
def generic(self, args, kws):
assert not kws
assert len(args) in [4, 5]
return signature(mpi_req_numba_type, *unliteral_all(args))
def generic(self, args, kws):
assert not kws
assert len(args) == 2
return signature(types.int32, *unliteral_all(args))
def generic(self, args, kws):
assert not kws
assert len(args) == 0
return signature(types.float64, *unliteral_all(args))
def generic(self, args, kws):
assert not kws
assert len(args) == 1
return signature(args[0], *unliteral_all(args))
def generic(self, args, kws):
assert not kws
assert len(args) == 2 # value and reduce_op
return signature(args[0], *unliteral_all(args))
def generic(self, args, kws):
assert not kws
assert len(args) == 2 # array and val
return signature(types.none, *unliteral_all(args))
def generic(self, args, kws):
arr = args[0] # array or series
# hiframes_typed pass replaces series input with array after typing
from sdc.hiframes.pd_series_ext import if_series_to_array_type
ret_typ = if_series_to_array_type(arr).copy(dtype=types.float64)
return signature(ret_typ, *unliteral_all(args))
