xobjects 0.4.2__tar.gz → 0.4.4__tar.gz

{xobjects-0.4.2/xobjects.egg-info → xobjects-0.4.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xobjects
-Version: 0.4.2
+Version: 0.4.4
 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.4/xobjects/_version.py

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

{xobjects-0.4.2 → xobjects-0.4.4}/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.2 → xobjects-0.4.4}/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.2 → xobjects-0.4.4}/xobjects/context_cupy.py

@@ -473,6 +473,9 @@ class ContextCupy(XContext):
 
         return out_kernels
 
+    def __str__(self):
+        return f"{type(self).__name__}:{cupy.cuda.get_device_id()}"
+
     def nparray_to_context_array(self, arr):
         """
         Copies a numpy array to the device memory.
@@ -631,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.2 → xobjects-0.4.4}/xobjects/context_pyopencl.py

@@ -218,7 +218,9 @@ class ContextPyopencl(XContext):
             with open(save_source_as, "w") as fid:
                 fid.write(specialized_source)
 
-        prg = cl.Program(self.context, specialized_source).build()
+        prg = cl.Program(self.context, specialized_source).build(
+            options="-cl-std=CL2.0",
+        )
 
         out_kernels = {}
         for pyname, kernel in kernel_descriptions.items():
@@ -236,6 +238,11 @@ class ContextPyopencl(XContext):
 
         return out_kernels
 
+    def __str__(self):
+        platform_id = cl.get_platforms().index(self.platform)
+        device_id = self.platform.get_devices().index(self.device)
+        return f"{type(self).__name__}:{platform_id}.{device_id}"
+
     def nparray_to_context_array(self, arr):
         """
         Copies a numpy array to the device memory.

{xobjects-0.4.2 → xobjects-0.4.4/xobjects.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xobjects
-Version: 0.4.2
+Version: 0.4.4
 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.2/xobjects/_version.py

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