xobjects 0.6.2__tar.gz → 0.6.3__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (57) hide show
  1. {xobjects-0.6.2/xobjects.egg-info → xobjects-0.6.3}/PKG-INFO +1 -1
  2. xobjects-0.6.3/xobjects/_version.py +1 -0
  3. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects/context_cupy.py +9 -4
  4. {xobjects-0.6.2 → xobjects-0.6.3/xobjects.egg-info}/PKG-INFO +1 -1
  5. xobjects-0.6.2/xobjects/_version.py +0 -1
  6. {xobjects-0.6.2 → xobjects-0.6.3}/Architecture.md +0 -0
  7. {xobjects-0.6.2 → xobjects-0.6.3}/LICENSE +0 -0
  8. {xobjects-0.6.2 → xobjects-0.6.3}/MANIFEST.in +0 -0
  9. {xobjects-0.6.2 → xobjects-0.6.3}/README.md +0 -0
  10. {xobjects-0.6.2 → xobjects-0.6.3}/pyproject.toml +0 -0
  11. {xobjects-0.6.2 → xobjects-0.6.3}/setup.cfg +0 -0
  12. {xobjects-0.6.2 → xobjects-0.6.3}/setup.py +0 -0
  13. {xobjects-0.6.2 → xobjects-0.6.3}/tests/test_align.py +0 -0
  14. {xobjects-0.6.2 → xobjects-0.6.3}/tests/test_array.py +0 -0
  15. {xobjects-0.6.2 → xobjects-0.6.3}/tests/test_buffer.py +0 -0
  16. {xobjects-0.6.2 → xobjects-0.6.3}/tests/test_capi.py +0 -0
  17. {xobjects-0.6.2 → xobjects-0.6.3}/tests/test_chunk.py +0 -0
  18. {xobjects-0.6.2 → xobjects-0.6.3}/tests/test_common.py +0 -0
  19. {xobjects-0.6.2 → xobjects-0.6.3}/tests/test_context_opencl.py +0 -0
  20. {xobjects-0.6.2 → xobjects-0.6.3}/tests/test_hybrid_class.py +0 -0
  21. {xobjects-0.6.2 → xobjects-0.6.3}/tests/test_kernel.py +0 -0
  22. {xobjects-0.6.2 → xobjects-0.6.3}/tests/test_linked_array.py +0 -0
  23. {xobjects-0.6.2 → xobjects-0.6.3}/tests/test_nplike_arrays.py +0 -0
  24. {xobjects-0.6.2 → xobjects-0.6.3}/tests/test_ref.py +0 -0
  25. {xobjects-0.6.2 → xobjects-0.6.3}/tests/test_scalars.py +0 -0
  26. {xobjects-0.6.2 → xobjects-0.6.3}/tests/test_shared_memory.py +0 -0
  27. {xobjects-0.6.2 → xobjects-0.6.3}/tests/test_strides.py +0 -0
  28. {xobjects-0.6.2 → xobjects-0.6.3}/tests/test_string.py +0 -0
  29. {xobjects-0.6.2 → xobjects-0.6.3}/tests/test_struct.py +0 -0
  30. {xobjects-0.6.2 → xobjects-0.6.3}/tests/test_to_dict.py +0 -0
  31. {xobjects-0.6.2 → xobjects-0.6.3}/tests/test_typeutils.py +0 -0
  32. {xobjects-0.6.2 → xobjects-0.6.3}/tests/test_unionref.py +0 -0
  33. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects/__init__.py +0 -0
  34. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects/_patch_pyopencl_array.py +0 -0
  35. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects/array.py +0 -0
  36. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects/capi.py +0 -0
  37. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects/context.py +0 -0
  38. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects/context_cpu.py +0 -0
  39. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects/context_pyopencl.py +0 -0
  40. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects/general.py +0 -0
  41. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects/headers/atomicadd.h +0 -0
  42. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects/headers/common.h +0 -0
  43. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects/hybrid_class.py +0 -0
  44. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects/linkedarray.py +0 -0
  45. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects/ref.py +0 -0
  46. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects/scalar.py +0 -0
  47. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects/specialize_source.py +0 -0
  48. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects/string.py +0 -0
  49. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects/struct.py +0 -0
  50. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects/test_helpers.py +0 -0
  51. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects/typeutils.py +0 -0
  52. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects/union.py +0 -0
  53. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects.egg-info/SOURCES.txt +0 -0
  54. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects.egg-info/dependency_links.txt +0 -0
  55. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects.egg-info/entry_points.txt +0 -0
  56. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects.egg-info/requires.txt +0 -0
  57. {xobjects-0.6.2 → xobjects-0.6.3}/xobjects.egg-info/top_level.txt +0 -0
{xobjects-0.6.2/xobjects.egg-info → xobjects-0.6.3}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: xobjects
3
- Version: 0.6.2
3
+ Version: 0.6.3
4
4
  Summary: In-memory serialization and code generator for CPU and GPU
5
5
  Home-page: https://xsuite.readthedocs.io/
6
6
  Download-URL: https://pypi.python.org/pypi/xobjects
xobjects-0.6.3/xobjects/_version.py ADDED
@@ -0,0 +1 @@
1
+ __version__ = "0.6.3"
{xobjects-0.6.2 → xobjects-0.6.3}/xobjects/context_cupy.py RENAMED
@@ -4,6 +4,7 @@
4
4
  # ########################################### #
5
5
 
6
6
  import logging
7
+ import warnings
7
8
  from typing import Dict, List, Tuple
8
9
 
9
10
  import numpy as np
@@ -32,6 +33,7 @@ try:
32
33
  import cupyx.scipy.special
33
34
  import cupyx.scipy.stats
34
35
  from cupyx.scipy import fftpack as cufftp
36
+ from cupy_backends.cuda.libs import nvrtc
35
37
 
36
38
  _enabled = True
37
39
  except ImportError:
@@ -40,6 +42,7 @@ except ImportError:
40
42
  message=("cupy is not installed. " "ContextCupy is not available!")
41
43
  )
42
44
  cufftp = cupy
45
+ nvrtc = None
43
46
  _enabled = False
44
47
 
45
48
  if _enabled:
@@ -466,12 +469,14 @@ class ContextCupy(XContext):
466
469
  *extra_compile_args,
467
470
  *include_flags,
468
471
  "-DXO_CONTEXT_CUDA",
469
- # Skip heavy optimizations (e.g. involving cloning),
470
- # which for us don't translate to a lot of runtime gains,
471
- # but consume a lot of compile time and memory:
472
- "--Ofast-compile=min",
473
472
  )
474
473
 
474
+ if nvrtc and nvrtc.getVersion() >= (12, 9):
475
+ # If supported, skip prohibitively heavy optimisations (e.g.
476
+ # involving cloning). This it at the expense of <20%
477
+ # runtime performance, but gain of a lot of compile time and memory.
478
+ extra_compile_args += ("--Ofast-compile=min",)
479
+
475
480
  module = cupy.RawModule(
476
481
  code=specialized_source, options=extra_compile_args
477
482
  )
{xobjects-0.6.2 → xobjects-0.6.3/xobjects.egg-info}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: xobjects
3
- Version: 0.6.2
3
+ Version: 0.6.3
4
4
  Summary: In-memory serialization and code generator for CPU and GPU
5
5
  Home-page: https://xsuite.readthedocs.io/
6
6
  Download-URL: https://pypi.python.org/pypi/xobjects
xobjects-0.6.2/xobjects/_version.py DELETED
@@ -1 +0,0 @@
1
- __version__ = "0.6.2"
{xobjects-0.6.2 → xobjects-0.6.3}/Architecture.md RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/LICENSE RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/MANIFEST.in RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/README.md RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/pyproject.toml RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/setup.cfg RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/setup.py RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/tests/test_align.py RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/tests/test_array.py RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/tests/test_buffer.py RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/tests/test_capi.py RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/tests/test_chunk.py RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/tests/test_common.py RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/tests/test_kernel.py RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/tests/test_ref.py RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/tests/test_scalars.py RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/tests/test_strides.py RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/tests/test_string.py RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/tests/test_struct.py RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/tests/test_to_dict.py RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/xobjects/__init__.py RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/xobjects/array.py RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/xobjects/capi.py RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/xobjects/context.py RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/xobjects/general.py RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/xobjects/ref.py RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/xobjects/scalar.py RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/xobjects/string.py RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/xobjects/struct.py RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/xobjects/typeutils.py RENAMED
File without changes
{xobjects-0.6.2 → xobjects-0.6.3}/xobjects/union.py RENAMED
File without changes