xobjects 0.6.1__tar.gz → 0.6.3__tar.gz

{xobjects-0.6.1/xobjects.egg-info → xobjects-0.6.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xobjects
-Version: 0.6.1
+Version: 0.6.3
 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.6.3/xobjects/_version.py

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

{xobjects-0.6.1 → xobjects-0.6.3}/xobjects/context_cupy.py

@@ -4,6 +4,7 @@
 # ########################################### #
 
 import logging
+import warnings
 from typing import Dict, List, Tuple
 
 import numpy as np
@@ -32,6 +33,7 @@ try:
     import cupyx.scipy.special
     import cupyx.scipy.stats
     from cupyx.scipy import fftpack as cufftp
+    from cupy_backends.cuda.libs import nvrtc
 
     _enabled = True
 except ImportError:
@@ -40,6 +42,7 @@ except ImportError:
         message=("cupy is not installed. " "ContextCupy is not available!")
     )
     cufftp = cupy
+    nvrtc = None
     _enabled = False
 
 if _enabled:
@@ -468,6 +471,12 @@ class ContextCupy(XContext):
             "-DXO_CONTEXT_CUDA",
         )
 
+        if nvrtc and nvrtc.getVersion() >= (12, 9):
+            # If supported, skip prohibitively heavy optimisations (e.g.
+            # involving cloning). This it at the expense of <20%
+            # runtime performance, but gain of a lot of compile time and memory.
+            extra_compile_args += ("--Ofast-compile=min",)
+
         module = cupy.RawModule(
             code=specialized_source, options=extra_compile_args
         )

{xobjects-0.6.1 → xobjects-0.6.3/xobjects.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xobjects
-Version: 0.6.1
+Version: 0.6.3
 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.6.1/xobjects/_version.py

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