xobjects 0.4.5__tar.gz → 0.4.7__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xobjects
-Version: 0.4.5
+Version: 0.4.7
 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-0.4.7}/tests/test_struct.py

@@ -3,6 +3,7 @@
 # Copyright (c) CERN, 2021.                   #
 # ########################################### #
 import cffi
+import numpy as np
 
 import xobjects as xo
 from xobjects.test_helpers import for_all_test_contexts, requires_context
@@ -213,6 +214,92 @@ def test_copy_dynamic():
     assert s2.b[1] == 4
 
 
+def test_kernel_namings():
+    class MyStruct(xo.Struct):
+        n = xo.Int32
+        var_mult_1 = xo.Float64[:]
+        var_mult_2 = xo.Float64[:]
+        var_mult_3 = xo.Float64[:]
+        var_mult_4 = xo.Float64[:]
+
+        _extra_c_sources = [
+            r"""
+double mul(MyStruct stru) {
+    int32_t n = MyStruct_get_n(stru);
+    double* var_mult_1 = MyStruct_getp1_var_mult_1(stru, 0);
+    double* var_mult_2 = MyStruct_getp1_var_mult_2(stru, 0);
+    double y = 0;
+    for (int32_t tid=0; tid<n; tid++){
+        y+= var_mult_1[tid] * var_mult_2[tid];
+        }
+    return y;
+    }
+
+double mult(MyStruct stru) {
+    int32_t n = MyStruct_get_n(stru);
+    double* var_mult_1 = MyStruct_getp1_var_mult_1(stru, 0);
+    double* var_mult_2 = MyStruct_getp1_var_mult_2(stru, 0);
+    double* var_mult_3 = MyStruct_getp1_var_mult_3(stru, 0);
+    double y = 0;
+    for (int32_t tid=0; tid<n; tid++){
+        y+= var_mult_1[tid] * var_mult_2[tid] * var_mult_3[tid];
+        }
+    return y;
+    }
+
+double mult_four(MyStruct stru) {
+    int32_t n = MyStruct_get_n(stru);
+    double* var_mult_1 = MyStruct_getp1_var_mult_1(stru, 0);
+    double* var_mult_2 = MyStruct_getp1_var_mult_2(stru, 0);
+    double* var_mult_3 = MyStruct_getp1_var_mult_3(stru, 0);
+    double* var_mult_4 = MyStruct_getp1_var_mult_4(stru, 0);
+    double y = 0;
+    for (int32_t tid=0; tid<n; tid++){
+        y+= var_mult_1[tid] * var_mult_2[tid] * var_mult_3[tid] * var_mult_4[tid];
+        }
+    return y;
+    }"""
+        ]
+
+    kernel_descriptions = {
+        "mul": xo.Kernel(
+            args=[xo.Arg(MyStruct, name="stru")],
+            ret=xo.Arg(xo.Float64),
+        ),
+        "mult": xo.Kernel(
+            c_name="mult",
+            args=[xo.Arg(MyStruct, name="stru")],
+            ret=xo.Arg(xo.Float64),
+        ),
+        "pyname_mul": xo.Kernel(
+            c_name="mult_four",
+            args=[xo.Arg(MyStruct, name="stru")],
+            ret=xo.Arg(xo.Float64),
+        ),
+    }
+
+    a1 = np.arange(10.0)
+    a2 = np.arange(10.0)
+    a3 = np.arange(10.0)
+    a4 = np.arange(10.0)
+    stru = MyStruct(
+        n=10, var_mult_1=a1, var_mult_2=a2, var_mult_3=a3, var_mult_4=a4
+    )
+
+    ctx = stru._context
+    ctx.add_kernels(kernels=kernel_descriptions)
+    assert "mul" in ctx.kernels
+    # assert "mult" in ctx.kernels
+    assert "pyname_mul" in ctx.kernels
+
+    y = ctx.kernels.mul(stru=stru)
+    assert y == 285.0
+    y = ctx.kernels.mult(stru=stru)
+    assert y == 2025.0
+    y = ctx.kernels.pyname_mul(stru=stru)
+    assert y == 15333.0
+
+
 @requires_context("ContextCpu")
 def test_compile_kernels_only_if_needed(tmp_path, mocker):
     """Test the use case of xtrack.

xobjects-0.4.7/xobjects/_version.py

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

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

@@ -398,7 +398,8 @@ class ContextCpu(XContext):
             _print("Compiling ContextCpu kernels...")
 
         for pyname, kernel in kernel_descriptions.items():
-            if pyname not in cdefs:  # check if kernel not already declared
+            # check if kernel not already declared
+            if f" {kernel.c_name}(" not in cdefs:
                 signature = cdef_from_kernel(kernel, pyname)
                 ffi_interface.cdef(signature)
                 log.debug(f"cffi def {pyname} {signature}")

{xobjects-0.4.5 → xobjects-0.4.7}/xobjects/test_helpers.py

@@ -39,11 +39,13 @@ def _for_all_test_contexts_excluding(
             f"this test: {excluding}."
         )(actual_test)
 
-    return pytest.mark.parametrize(
+    test = pytest.mark.parametrize(
         "test_context",
         test_context_names,
     )(actual_test)
 
+    return pytest.mark.context_dependent(test)
+
 
 def for_all_test_contexts(*args, **kwargs):
     """Parametrize the decorated test over all test contexts with the argument
@@ -80,6 +82,26 @@ def requires_context(context_name: str):
     ctx_names = (type(ctx).__name__ for ctx in get_test_contexts())
 
     if {context_name} & set(ctx_names):  # proceed as normal
-        return lambda test_function: test_function
+        return pytest.mark.context_dependent
 
     return pytest.mark.skip(f"{context_name} is unavailable on this platform.")
+
+
+def fix_random_seed(seed: int):
+    """Decorator to fix the random seed for a test."""
+
+    def decorator(test_function):
+        @wraps(test_function)
+        def wrapper(*args, **kwargs):
+            import numpy as np
+
+            rng_state = np.random.get_state()
+            try:
+                np.random.seed(seed)
+                test_function(*args, **kwargs)
+            finally:
+                np.random.set_state(rng_state)
+
+        return wrapper
+
+    return decorator

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xobjects
-Version: 0.4.5
+Version: 0.4.7
 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-0.4.7}/xobjects.egg-info/SOURCES.txt

@@ -1,48 +1,7 @@
-.gitignore
-.pre-commit-config.yaml
-.readthedocs.yaml
-Architecture.md
 LICENSE
-Readme.md
-contributors_and_credits.txt
-mktest
 pyproject.toml
-release.sh
-requirements.txt
 setup.cfg
 setup.py
-update_cprght_statement.py
-.github/workflows/black.yml
-docs/Makefile
-docs/architecture.rst
-docs/conf.py
-docs/index.rst
-docs/make.bat
-docs/quickstart.rst
-docs/reference.rst
-docs/architecture/contexts.rst
-docs/architecture/types.rst
-examples/api_def.py
-examples/api_init.py
-examples/capi_example.c
-examples/check_assembly.c
-examples/ex_array_init_perf.py
-examples/ex_cpugpu.py
-examples/ex_nested_ref.py
-examples/ex_ref_cpu.py
-examples/ex_struct_cpu.py
-examples/ex_unionref_cpu.py
-examples/ex_unionref_method.py
-examples/init_soa.py
-examples/issue_with_two_pyopencl_contexts.py
-examples/opencl_offsets.py
-examples/simple_kernel.py
-examples/sixtracklib.py
-examples/kernel_basics/kernel_cffi.py
-examples/kernel_basics/kernel_cupy.py
-examples/kernel_basics/kernel_pyopencl.py
-tests/__init__.py
-tests/notest_capi.py
 tests/test_align.py
 tests/test_array.py
 tests/test_buffer.py
@@ -62,7 +21,6 @@ tests/test_struct.py
 tests/test_to_json.py
 tests/test_typeutils.py
 tests/test_unionref.py
-tests/__pycache__/test_context.cpython-37-pytest-6.2.1.pyc
 xobjects/__init__.py
 xobjects/_patch_pyopencl_array.py
 xobjects/_version.py

xobjects-0.4.5/.github/workflows/black.yml

@@ -1,10 +0,0 @@
-name: Lint
-
-on: [push, pull_request]
-
-jobs:
-  lint:
-    runs-on: ubuntu-latest
-    steps:
-      - uses: actions/checkout@v2
-      - uses: psf/black@stable

xobjects-0.4.5/.gitignore

@@ -1,11 +0,0 @@
-__pycache__/
-*.pyc
-*.pyo
-*.swp
-*.egg-info/
-docs/_build
-cov_html
-.vscode
-.pytest_cache
-.coverage
-*.c

xobjects-0.4.5/.pre-commit-config.yaml

@@ -1,12 +0,0 @@
-repos:
-  - repo: https://github.com/pre-commit/pre-commit-hooks
-    rev: v4.5.0
-    hooks:
-    -   id: check-yaml
-    -   id: end-of-file-fixer
-    -   id: trailing-whitespace
-  - repo: https://github.com/psf/black
-    rev: 23.10.1
-    hooks:
-      - id: black
-        language_version: python3 # Should be a command that runs python3.6+

xobjects-0.4.5/.readthedocs.yaml

@@ -1,20 +0,0 @@
-# .readthedocs.yaml
-# Read the Docs configuration file
-# See https://docs.readthedocs.io/en/stable/config-file/v2.html for details
-
-# Required
-version: 2
-
-# Build documentation in the docs/ directory with Sphinx
-sphinx:
-   configuration: docs/conf.py
-
-# Optionally build your docs in additional formats such as PDF
-formats:
-   - pdf
-
-# Optionally set the version of Python and requirements required to build your docs
-python:
-   version: 3.7
-#   install:
-#   - requirements: docs/requirements.txt

xobjects-0.4.5/Architecture.md

@@ -1,202 +0,0 @@
-# Architecture
-
-## Introduction
-
-The library is based on:
-- Buffers: manage chunck of bytes on CPU and GPU (Cuda and OpenCL) memory using different backends
-- Contexts: keep track of buffers, launch functions operating on buffers
-- Basic types: a set of basic types that allow to describe structured data. Basic types are scalars, strings, structs, arrays and unions. User defined classes, composed with basic types, allow to interact with data on buffers and generate C functions to operate with data.
-- C-API generation: The type information allows to write C functions that can be interact with data. The C functions are specialized for CPU, OpenCL and Cuda. Also loops described by annotation in comments are automatically generated for CPU, OpenCL and Cuda.
-
-When a user defined class is created, Python methods and additional metadeta is attached to the class.
-When an instance of a user defined class is instantiated, memory is written to a buffer and a Python object is created to interact with the data.
-The size of an instance may not be known at class defintion time, but it cannot be changed after initialization.
-
-## Contexts and Buffers
-
-A context contain:
-
-- `new_buffer(capacity)`: create new buffer of initial capacity
-- `buffers`: list of weakref to buffers
--  kernels : Object containing compiled code from sources/
-
-Buffer:
-- `buffer`: low level buffer
-- `allocate(self,size)`: return the offset of the first free memory chunk of given size and grows the buffer if necessary
-- `free(self,offset,size)`: free a chuck of memory of a given size
-- `grow(self,capacity)`: increase the size of a buffer
-- `_new_buffer(capacity)`: creae lowlevel buffer
-- `update_from_native(...)`: Copy data from native buffer into self.buffer starting from offset
-- `update_from_buffer(...)`: Copy data from python buffer such as bytearray, bytes, memoryview, numpy array.data
-- `update_from_nplike(...)`: Copy data from nplike matching dest_dtype
-- `update_from_xbuffer(...)`: Copy data from any xbuffer, don't pass through cpu if possible
-- `copy_to_native(...)`: Copy data from self.buffer into dest
-- `to_native(...)`: Return native data with content at from offset and nbytes
-- `to_nplike(...)`: Return a view in of an nplike
-- `to_bytearray(...)`: Return a byte array: used in update_from_xbuffer
-- `to_pointer_arg(...)`: Return data that can be used as argument in kernel
-
-## Types
-
-Types can be composed of:
-- scalar: numbers, String
-- compound: Struct, Array, Ref, UnionRef 
-
-### Scalars
-- examples: Float64, Int64, ...
-- create: Float64(3.14)
-- memory layout
-    - data 
-
-### String:
-- create: String(string_or_int)
-- memory layout
-    - size
-    - data 
-
-
-### Struct
-- example: struct MyStruct field1 field2 ...
-- create: Struct(dict), Struct(**args)
-- memory layout:
-  - [ instance size ]
-  - static-field1
-  -  ..
-  - static-fieldn
-  - [ offset field 2 ]
-  - [ offset ...     ]
-  - [ offset field n ]
-  - [ dynamic-field1 ]
-  - [ ... ]
-  - [ dynamic-fieldn ]
-
-### Array:
-- example: 'array' 'f64' d1 d2 d3 'C' ; 'array' 'i64' d1 ':' d3 'F' ;
-- where:
-   - 'C'/'F' is C-contiguos or F-contiguous
-   - d1, d2, .. are the dimensions for each axis and ':' means not specified
-- create: Array(d1,d2,...) or Array([...]) or Array(np-array)
-- memory layout:
-  - [size] if not _is_static_shape or not _is_static_type
-  - [dims ... ] len(_dynamic_shape)
-  - [strides...] if nd>1 and dynamic shapes
-  - [offsets] if not _is_static_type
-  - data
-
-### Union:
-- create: Union(obj), Union( (typename,args) )
-- memory layout
-  - typeid
-  - data
-
-### UnionRef:
-- create: UnionRef(obj), UnionRef( (typename,args) )
-- memory layout
-  - typeid
-  - offset
-
-
-### Ref:
-- memory layout
-  - offset
-
-## Implementation details
-
-Assumptions:
-- Python user API cannot change the internal structure of the object but only the content
-- Python object caches the structure information to avoid round trips to access data
-- C user API does not manipulate structures
-
-
-Operation:
-
-- _inspect_args(*args, **args) -> info: Info
-  - check value and calculate size, offsets and dimensions from arguments, recursively
-  - returns at least `size` for `_get_a_buffer` and `value` for `_to_buffer`
-
-- _get_a_buffer(size, _context, _buffer, _offset):
-  - make sure a valid buffer and offset is created
-
-- _to_buffer(buffer, offset, value, info, size?):
-  - set data on buffer with offset from pyhon value, using info and implicitely respecting size
-  - if value same class -> binary copy (passing from context if needed)
-
-- _from_buffer(buffer, offset) -> value:
-  - create python object from data on buffer. Can be a native type (scalar or string) or another XObject (struct, array)
-
-
-- __init__(self,*args,_context=None, _buffer=None, _offset=None,*nargs)
-  - initialize object data on buffer from args and create python object:
-    - use _pre_init(*args,**nargs) -> return args, nargs standard initialization value
-    - check values and calculate size using `_inspect_args`, return info object containing size, offsets and values to be later used by _to_buffer
-    - allocate using `_get_a_buffer`
-    - write data using `_to_buffer`
-    - use `_post_init` to complete object
-`
-- _update(self, value, size=None):
-  - Optional update object content using value, in case respecting size for string
-
-- __get__(field,instance) or __getitem__(self,index...)
-  - if return instance._cache[field.index] else  #should implement item caching for struct and array
-  - else return _from_buffer
-
-- __set__(..., value) or __setitem__(...,value)
-  - if hasattr(self.ftype._update) get object and update in place
-    else call _to_buffer
-
-- json enconding
-  nested object can be initialized using a json enconding
-  d:dict -> type(**d)
-  t:tuple -> type(*t)
-  a:other -> type(a)
-
-
-
-Class content:
-
-- _inspect_args(cls, *args, **nargs) -> size, offsets
-  compute size and sizes and offsets from arguments
-
-- _to_buffer(cls, buffer, offset, value, offsets=None):
-  initialize object on buffer
-
-- _from_buffer(cls, buffer,offset):
-  create python object from initlialized object on buffer,offset
-
-- _size: object size
-  None if dynamic
-
-Instance content:
-
-- _get_size(self) -> size
-
-## C-API
-
-Access
-- `<Class>_get<_attrs>(obj, <indexes>)` -> scalar or object pointer
-- `<Class>_set<_attrs>(obj, <indexes>, scalar)`
-
-e.g. Multipole_get_field_normal(3)
-
-Kernel convention:
-
-```
-<fun>(void **refs){
-          Type0 arg0 = (Type0) refs[0];
-          ....
-          Type1 argN = (TypeN) refs[N];
-        ...
-        }
-```
-
-``` <fun>(buffer1, offset, ..., offset3){
-         Type0 arg0 = (Type0) refs[offset0];
-         ,,,
-         TypeN argN = (TypeN) refs[offsetN];
-}
-```
-
-## TODO
-
-- add free methods to delete data from buffer
-- implement SOA types

xobjects-0.4.5/Readme.md

@@ -1,10 +0,0 @@
-# XObjects
-
-Library to create and manipulate serialized object in CPU and GPU memory efficiently.
-
-The library defines:
--  Contexts: create buffers and manage computations e.g. BytearrayContext, CLContext, ...
--  Buffers: reference to internal memory buffer with an allocator e.g.
-   BytearrayContext, CLBuffer, ...
--  Data types: define users define types to create and manipulate objects on
-   buffers and generate C api, e.g. Float64, Int64, String, Struct, Array, Union

xobjects-0.4.5/contributors_and_credits.txt

@@ -1,7 +0,0 @@
-The following people contributed to the development of this package:
-
-CERN contributors:
- - R. De Maria
- - G. Iadarola
- - P. Kiksiny
- - S. Lopaciuk

xobjects-0.4.5/docs/Makefile

@@ -1,20 +0,0 @@
-# Minimal makefile for Sphinx documentation
-#
-
-# You can set these variables from the command line, and also
-# from the environment for the first two.
-SPHINXOPTS    ?=
-SPHINXBUILD   ?= sphinx-build
-SOURCEDIR     = .
-BUILDDIR      = _build
-
-# Put it first so that "make" without argument is like "make help".
-help:
-	@$(SPHINXBUILD) -M help "$(SOURCEDIR)" "$(BUILDDIR)" $(SPHINXOPTS) $(O)
-
-.PHONY: help Makefile
-
-# Catch-all target: route all unknown targets to Sphinx using the new
-# "make mode" option.  $(O) is meant as a shortcut for $(SPHINXOPTS).
-%: Makefile
-	@$(SPHINXBUILD) -M $@ "$(SOURCEDIR)" "$(BUILDDIR)" $(SPHINXOPTS) $(O)