numbox 0.1.3__py3-none-any.whl → 0.1.4__py3-none-any.whl

numbox/__init__.py

@@ -1 +1 @@
-__version__ = '0.1.3'
+__version__ = '0.1.4'

numbox/core/any_type.py

@@ -1,6 +1,8 @@
+from inspect import getmodule, getsource
 from numba import njit
 from numba.core.errors import NumbaError
-from numba.core.types import StructRef
+from numba.core.itanium_mangler import mangle_type_or_value
+from numba.core.types import StructRef, Type
 from numba.experimental.structref import define_boxing, new, register, StructRefProxy
 from numba.extending import overload, overload_method
 
@@ -25,7 +27,10 @@ class _Content:
     pass
 
 
-@overload(_Content, strict=False)
+default_jit_options = {'cache': True}
+
+
+@overload(_Content, strict=False, jit_options=default_jit_options)
 def ol_content(x_ty):
     x_ty = prune_type(x_ty)
     content_type = ContentTypeClass([("x", x_ty)])
@@ -49,11 +54,11 @@ class Any(StructRefProxy):
     def __new__(cls, x):
         raise NotImplementedError(deleted_any_ctor_error)
 
-    @njit
+    @njit(cache=True)
     def get_as(self, ty):
         return self.get_as(ty)
 
-    @njit
+    @njit(cache=True)
     def reset(self, val):
         return self.reset(val)
 
@@ -67,22 +72,45 @@ define_boxing(AnyTypeClass, Any)
 AnyType = AnyTypeClass([("p", ErasedType)])
 
 
-@overload_method(AnyTypeClass, "get_as", strict=False)
+@overload_method(AnyTypeClass, "get_as", strict=False, jit_options=default_jit_options)
 def ol_get_as(self_class, ty_class):
     def _(self, ty):
         return deref(self.p, ty)
     return _
 
 
-@overload_method(AnyTypeClass, "reset", strict=False)
+@overload_method(AnyTypeClass, "reset", strict=False, jit_options=default_jit_options)
 def ol_reset(self_class, x_class):
     def _(self, x):
         self.p = cast(_Content(x), ErasedType)
     return _
 
 
-@njit
-def make_any(x):
+def make_any_prototype(x):
     any = new(AnyType)
     any.p = cast(_Content(x), ErasedType)
     return any
+
+
+make_any = njit(**default_jit_options)(make_any_prototype)
+
+
+def make_any_maker(arg_type=None, jit_options=None):
+    if arg_type is None:
+        return make_any
+    if not isinstance(arg_type, Type):
+        raise TypeError(f"Expected numba type, got {arg_type}")
+    jit_options = {**default_jit_options, **(jit_options or {})}
+    sig = AnyType(arg_type)
+    custom_make_any_name = f"make_any_{mangle_type_or_value(arg_type)}"
+    source_code = getsource(make_any_prototype)
+    source_code = source_code.replace('make_any_prototype', custom_make_any_name)
+    code_txt = f"""
+@njit(sig, **jit_options)
+{source_code}
+"""
+    code = compile(code_txt, __file__, mode='exec')
+    ns = {**getmodule(make_any).__dict__, **{'sig': sig, 'jit_options': jit_options}}
+    exec(code, ns)
+    custom_make_any = ns[custom_make_any_name]
+    return custom_make_any

{numbox-0.1.3.dist-info → numbox-0.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: numbox
-Version: 0.1.3
+Version: 0.1.4
 Author: Mikhail Goykhman
 License: MIT License (with Citation Clause)
         

numbox-0.1.4.dist-info/RECORD

@@ -0,0 +1,13 @@
+numbox/__init__.py,sha256=aBEbDvx4LMg8A1TJJR6dEHu8rQODVin528hLS_EDvuA,22
+numbox/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+numbox/core/any_type.py,sha256=qi5vZD5-s-3ksTORes4rCF2t7H6jpz3whsKNxc4Oyq8,2843
+numbox/core/meminfo.py,sha256=-b7wlCRyqOZjvKjEuwNwzqE4XvnEWxu3I2k4DEP3Cqk,1755
+numbox/core/proxy.py,sha256=kGYlEdLK40lxxu56e_S32s9YuQ6AuQnFegt5uQrUw5w,3889
+numbox/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+numbox/utils/highlevel.py,sha256=hAmF2y2ayPJT0qpwHrGDt09R4-6Ts9sTkCTh-8FTIUU,319
+numbox/utils/lowlevel.py,sha256=6CRfFbMeBZGM7hqPAQzyClzU_e_IsNpqONJknFWGzJM,1405
+numbox-0.1.4.dist-info/LICENSE,sha256=YYgNvjH_p6-1NsdrIqGJnr1GUbZzA_8DxsP6vVfM6nY,1446
+numbox-0.1.4.dist-info/METADATA,sha256=kx51gPJjDouxGwrmuwfijb72klqlmQ5jC8IzVg1suTw,2271
+numbox-0.1.4.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+numbox-0.1.4.dist-info/top_level.txt,sha256=A67jOkfqidCSYYm6ifjN_WZyIiR1B27fjxv6nNbPvjc,7
+numbox-0.1.4.dist-info/RECORD,,