Merge branch 'topic/default/pythran-type' into 'branch/default'

Support for Pythran "type"

Closes #60

See merge request fluiddyn/transonic!159

Author: paugier

data_tests/literal_with_types.py

@@ -0,0 +1,6 @@
+from transonic import boost, Literal
+
+
+@boost
+def gen_seq(dtype: Literal[int, float], seq_type: Literal[list, set], n: int):
+    return seq_type(map(dtype, range(n)))

data_tests/saved__backend__/cython/default_params.pxd

@@ -3,7 +3,7 @@ import cython
 import numpy as np
 cimport numpy as np
 
-ctypedef fused __func__Unionstr_None:
+ctypedef fused __func__Union_str_None:
    cython.str
 
-cpdef func(cython.int a=*, __func__Unionstr_None b=*, cython.float c=*)
+cpdef func(cython.int a=*, __func__Union_str_None b=*, cython.float c=*)

data_tests/saved__backend__/cython/literal_with_types.pxd

@@ -0,0 +1,12 @@
+import cython
+
+import numpy as np
+cimport numpy as np
+
+ctypedef fused __gen_seq__Literal_int_float:
+   object
+
+ctypedef fused __gen_seq__Literal_list_set:
+   object
+
+cpdef gen_seq(__gen_seq__Literal_int_float dtype, __gen_seq__Literal_list_set seq_type, cython.int n)

data_tests/saved__backend__/cython/literal_with_types.py

@@ -0,0 +1,12 @@
+try:
+    import cython
+except ImportError:
+    from transonic_cl import cython
+
+
+def gen_seq(dtype, seq_type, n):
+    return seq_type(map(dtype, range(n)))
+
+
+def __transonic__():
+    return "0.7.3"

data_tests/saved__backend__/cython/type_hint_notemplate.pxd

@@ -13,12 +13,12 @@ ctypedef fused __compute__TypeIint_complex128I:
    cython.int
    np.complex128_t
 
-ctypedef fused __compute__UnionArray_TypeIint_complex128I_NDimI1_3I_Array_float32_NDimI1_3Ip1:
+ctypedef fused __compute__Union_Array_TypeIint_complex128I_NDimI1_3I_Array_float32_NDimI1_3Ip1:
    np.ndarray[np.complex128_t, ndim=1]
    np.ndarray[np.complex128_t, ndim=3]
    np.ndarray[np.float32_t, ndim=2]
    np.ndarray[np.float32_t, ndim=4]
    np.ndarray[np.int32_t, ndim=1]
    np.ndarray[np.int32_t, ndim=3]
 
-cpdef compute(__compute__Array_TypeIint_complex128I_NDimI1_3I a, __compute__Array_TypeIint_complex128I_NDimI1_3I b, __compute__TypeIint_complex128I c, __compute__UnionArray_TypeIint_complex128I_NDimI1_3I_Array_float32_NDimI1_3Ip1 d, cython.str e)
+cpdef compute(__compute__Array_TypeIint_complex128I_NDimI1_3I a, __compute__Array_TypeIint_complex128I_NDimI1_3I b, __compute__TypeIint_complex128I c, __compute__Union_Array_TypeIint_complex128I_NDimI1_3I_Array_float32_NDimI1_3Ip1 d, cython.str e)

data_tests/saved__backend__/jax/literal_with_types.py

@@ -0,0 +1,10 @@
+# __protected__ from jax import jit
+# __protected__ @jit
+
+
+def gen_seq(dtype, seq_type, n):
+    return seq_type(map(dtype, range(n)))
+
+
+def __transonic__():
+    return "0.7.3"

data_tests/saved__backend__/numba/literal_with_types.py

@@ -0,0 +1,10 @@
+# __protected__ from numba import njit
+# __protected__ @njit(cache=True, fastmath=True)
+
+
+def gen_seq(dtype, seq_type, n):
+    return seq_type(map(dtype, range(n)))
+
+
+def __transonic__():
+    return "0.7.3"

data_tests/saved__backend__/python/literal_with_types.py

@@ -0,0 +1,6 @@
+def gen_seq(dtype, seq_type, n):
+    return seq_type(map(dtype, range(n)))
+
+
+def __transonic__():
+    return "0.7.3"

data_tests/saved__backend__/pythran/literal_with_types.py

@@ -0,0 +1,6 @@
+def gen_seq(dtype, seq_type, n):
+    return seq_type(map(dtype, range(n)))
+
+
+def __transonic__():
+    return "0.7.3"

data_tests/saved__backend__/pythran/literal_with_types.pythran

@@ -0,0 +1,6 @@
+export gen_seq(float64 type, list type, int)
+export gen_seq(float64 type, set type, int)
+export gen_seq(int type, list type, int)
+export gen_seq(int type, set type, int)
+
+export __transonic__()