xobjects 0.5.12__tar.gz → 0.5.13__tar.gz

{xobjects-0.5.12/xobjects.egg-info → xobjects-0.5.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xobjects
-Version: 0.5.12
+Version: 0.5.13
 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.5.13/xobjects/_version.py

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

{xobjects-0.5.12 → xobjects-0.5.13}/xobjects/context_cpu.py

@@ -316,6 +316,12 @@ class ContextCpu(XContext):
             if _forbid_compile:
                 raise RuntimeError("Compilation is forbidden")
 
+            if os.environ.get('XOBJECTS_FORBID_COMPILE'):
+                raise RuntimeError(
+                    "Compilation is forbidden by the environment variable "
+                    "XOBJECTS_FORBID_COMPILE"
+                )
+
             so_file = self.compile_kernel(
                 module_name,
                 kernel_descriptions,
@@ -376,8 +382,9 @@ class ContextCpu(XContext):
         )
         out_kernels = {}
         for pyname, kernel_desc in kernel_descriptions.items():
+            c_name = kernel_desc.c_name or pyname
             out_kernels[pyname] = KernelCpu(
-                function=getattr(module.lib, kernel_desc.c_name),
+                function=getattr(module.lib, c_name),
                 description=kernel_desc,
                 ffi_interface=module.ffi,
                 context=self,
@@ -459,14 +466,15 @@ class ContextCpu(XContext):
             return Path(output_file)
         finally:
             # Clean temp files
-            files_to_remove = [
-                module_name + ".c",
-                module_name + ".o",
-            ]
-
-            for ff in files_to_remove:
-                if os.path.exists(ff):
-                    os.remove(ff)
+            if 'XOBJECTS_KEEP_BUILD_FILES' not in os.environ:
+                files_to_remove = [
+                    module_name + ".c",
+                    module_name + ".o",
+                ]
+
+                for ff in files_to_remove:
+                    if os.path.exists(ff):
+                        os.remove(ff)
 
     def _build_sources(
         self,

{xobjects-0.5.12 → xobjects-0.5.13}/xobjects/struct.py

@@ -475,6 +475,32 @@ class Struct(metaclass=MetaStruct):
         extra_classes=(),
         extra_compile_args=(),
     ):
+
+        if context.allow_prebuilt_kernels:
+            _print_state = Print.suppress
+            Print.suppress = True
+            try:
+                from xsuite import (
+                    get_suitable_kernel,
+                    XSK_PREBUILT_KERNELS_LOCATION,
+                )
+                kernel_info = get_suitable_kernel(
+                    config={},
+                    tracker_element_classes=[],
+                    classes=list(extra_classes) + [cls],
+                )
+            except ImportError:
+                kernel_info = None
+
+            Print.suppress = _print_state
+            if kernel_info:
+                kernels = context.kernels_from_file(
+                    module_name=kernel_info['module_name'],
+                    containing_dir=XSK_PREBUILT_KERNELS_LOCATION,
+                    kernel_descriptions=cls._kernels,
+                )
+                context.kernels.update(kernels)
+
         if only_if_needed:
             all_found = True
             for kk, kernel_description in cls._kernels.items():

{xobjects-0.5.12 → xobjects-0.5.13}/xobjects/test_helpers.py

@@ -5,6 +5,7 @@
 
 from functools import wraps
 from typing import Callable, Iterable, Union
+import os
 
 import pytest
 
@@ -105,3 +106,10 @@ def fix_random_seed(seed: int):
         return wrapper
 
     return decorator
+
+def skip_if_forbid_compile():
+    if os.environ.get('XOBJECTS_FORBID_COMPILE'):
+        pytest.skip(
+            "Compilation is forbidden by the environment variable "
+            "XOBJECTS_FORBID_COMPILE"
+        )

{xobjects-0.5.12 → xobjects-0.5.13/xobjects.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xobjects
-Version: 0.5.12
+Version: 0.5.13
 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.5.12/xobjects/_version.py

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