Example #1
0
 def generic(self, args, kws):
     assert not kws
     assert len(args) == 4
     return signature(string_array_type, *unliteral_all(args))
Example #2
0
 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))
Example #3
0
 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))
Example #4
0
 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))
Example #5
0
 def generic(self, args, kws):
     assert not kws
     assert len(args) == 2  # array and val
     return signature(types.none, *unliteral_all(args))
Example #6
0
 def generic(self, args, kws):
     assert not kws
     assert len(args) == 2
     # args: str_arr, pat
     return signature(types.Array(types.boolean, 1, 'C'), *unliteral_all(args))
Example #7
0
 def generic(self, args, kws):
     assert not kws
     assert len(args) == 0
     return signature(types.float64, *unliteral_all(args))
Example #8
0
 def resolve_pop(self, dict, args, kws):
     assert not kws
     return signature(dict.val_typ, *unliteral_all(args))
Example #9
0
 def generic(self, args, kws):
     assert not kws
     assert len(args) == 2
     return signature(h5file_type, *unliteral_all(args))
Example #10
0
 def generic(self, args, kws):
     assert not kws
     assert len(args) == 2  # value and reduce_op
     return signature(args[0], *unliteral_all(args))
Example #11
0
 def resolve_create_dataset(self, f_id, args, kws):
     return _create_dataset_typer(unliteral_all(args), kws)
Example #12
0
 def resolve_create_group(self, f_id, args, kws):
     return signature(h5group_type, *unliteral_all(args))
Example #13
0
 def generic(self, args, kws):
     assert not kws
     assert len(args) == 2
     return signature(SeriesType(types.bool_, 1, 'C'), *unliteral_all(args))
Example #14
0
 def generic(self, args, kws):
     arr = args[0]  # array or series
     # hiframes_typed pass replaces series input with array after typing
     from hpat.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))
Example #15
0
 def generic(self, args, kws):
     assert not kws
     assert len(args) in [4, 5]
     return signature(mpi_req_numba_type, *unliteral_all(args))
Example #16
0
 def resolve_get(self, dict, args, kws):
     assert not kws
     assert len(args) == 2
     return signature(args[1], *unliteral_all(args))
Example #17
0
 def generic(self, args, kws):
     assert not kws
     assert len(args) == 2
     return signature(types.int32, *unliteral_all(args))
Example #18
0
 def generic(self, args, kws):
     if len(args) == 1 and isinstance(args[0], DictKeyIteratorType):
         return signature(args[0].key_typ, *unliteral_all(args))
Example #19
0
 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))
Example #20
0
 def generic(self, args, kws):
     assert not kws
     assert len(args) == 2  # array and count
     return signature(args[0], *unliteral_all(args))
Example #21
0
 def generic(self, args, kws):
     arr = args[0]  # array or series
     return signature(arr.copy(dtype=types.float64), *unliteral_all(args))