xobjects 0.4.3__tar.gz → 0.4.5__tar.gz

{xobjects-0.4.3 → xobjects-0.4.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xobjects
-Version: 0.4.3
+Version: 0.4.5
 Summary: In-memory serialization and code generator for CPU and GPU
 Home-page: https://xsuite.readthedocs.io/
 Download-URL: https://pypi.python.org/pypi/xobjects

xobjects-0.4.5/xobjects/_version.py

@@ -0,0 +1 @@
+__version__ = "0.4.5"

{xobjects-0.4.3 → xobjects-0.4.5}/xobjects/capi.py

@@ -471,7 +471,7 @@ def gen_method_switch(cls, path, conf, method):
         #endif"""
         )
     lst.append("  }")
-    lst.append("  return 0;")
+    lst.append(f"  return{'' if method.ret is None else ' 0'};")
     lst.append("}")
     return "\n".join(lst), kernel
 

{xobjects-0.4.3 → xobjects-0.4.5}/xobjects/context.py

@@ -210,19 +210,30 @@ class XContext(ABC):
 
     def __init__(self):
         self._kernels = KernelDict()
-        self._buffers = []
+        self._buffers = weakref.WeakSet()
+        self._allocations = 0
 
     def __str__(self):
         return type(self).__name__
 
     def new_buffer(self, capacity=1048576):
         buf = self._make_buffer(capacity=capacity)
-        self.buffers.append(weakref.finalize(buf, log.debug, "free buf"))
+        self._buffers.add(buf)
+        self._allocations += 1
         return buf
 
+    def __getstate__(self):
+        state = self.__dict__
+        del state["_buffers"]
+        return state
+
+    def __setstate__(self, state):
+        self.__dict__.update(state)
+        self._buffers = weakref.WeakSet()
+
     @property
     def buffers(self):
-        return self._buffers
+        return list(self._buffers)
 
     @property
     def kernels(self):

{xobjects-0.4.3 → xobjects-0.4.5}/xobjects/context_cpu.py

@@ -10,6 +10,7 @@ import sysconfig
 import uuid
 from pathlib import Path
 from typing import Callable, Dict, List, Sequence, Tuple
+import weakref
 
 from .general import _print
 
@@ -106,6 +107,9 @@ class LinkedArrayCpu(BaseLinkedArray, np.ndarray):
             order="C",
         )
 
+    def copy(self):
+        return np.array(self)
+
 
 def _so_for_module_name(name, containing_dir=".") -> Path:
     # The so file name is something like:
@@ -642,10 +646,12 @@ class ContextCpu(XContext):
     def __getstate__(self):
         state = self.__dict__.copy()
         state["_kernels"] = {}
+        del state["_buffers"]
         return state
 
     def __setstate__(self, state):
         self.__dict__.update(state)
+        self._buffers = weakref.WeakSet()
 
 
 class BufferByteArray(XBuffer):

{xobjects-0.4.3 → xobjects-0.4.5}/xobjects/context_cupy.py

@@ -634,7 +634,6 @@ class KernelCupy(object):
     def __init__(
         self, function, description, block_size, context, shared_mem_size_bytes
     ):
-
         self.function = function
         self.description = description
         self.block_size = block_size

{xobjects-0.4.3 → xobjects-0.4.5}/xobjects.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xobjects
-Version: 0.4.3
+Version: 0.4.5
 Summary: In-memory serialization and code generator for CPU and GPU
 Home-page: https://xsuite.readthedocs.io/
 Download-URL: https://pypi.python.org/pypi/xobjects

xobjects-0.4.3/xobjects/_version.py

@@ -1 +0,0 @@
-__version__ = "0.4.3"