numba-cuda 0.8.0__py3-none-any.whl → 0.9.0__py3-none-any.whl

numba_cuda/numba/cuda/cudamath.py

@@ -136,5 +136,5 @@ class Math_isnan(ConcreteTemplate):
 class Math_modf(ConcreteTemplate):
     cases = [
         signature(types.UniTuple(types.float64, 2), types.float64),
-        signature(types.UniTuple(types.float32, 2), types.float32)
+        signature(types.UniTuple(types.float32, 2), types.float32),
     ]

numba_cuda/numba/cuda/debuginfo.py

@@ -7,7 +7,6 @@ _BYTE_SIZE = 8
 
 
 class CUDADIBuilder(DIBuilder):
-
     def _var_type(self, lltype, size, datamodel=None):
         is_bool = False
         is_grid_group = False
@@ -34,11 +33,14 @@ class CUDADIBuilder(DIBuilder):
             elif is_grid_group:
                 ditok = "DW_ATE_unsigned"
 
-            return m.add_debug_info('DIBasicType', {
-                'name': name,
-                'size': bitsize,
-                'encoding': ir.DIToken(ditok),
-            })
+            return m.add_debug_info(
+                "DIBasicType",
+                {
+                    "name": name,
+                    "size": bitsize,
+                    "encoding": ir.DIToken(ditok),
+                },
+            )
 
         # For other cases, use upstream Numba implementation
         return super()._var_type(lltype, size, datamodel=datamodel)

numba_cuda/numba/cuda/decorators.py

@@ -6,13 +6,24 @@ from numba.cuda.dispatcher import CUDADispatcher
 from numba.cuda.simulator.kernel import FakeCUDAKernel
 
 
-_msg_deprecated_signature_arg = ("Deprecated keyword argument `{0}`. "
-                                 "Signatures should be passed as the first "
-                                 "positional argument.")
-
-
-def jit(func_or_sig=None, device=False, inline=False, link=[], debug=None,
-        opt=None, lineinfo=False, cache=False, **kws):
+_msg_deprecated_signature_arg = (
+    "Deprecated keyword argument `{0}`. "
+    "Signatures should be passed as the first "
+    "positional argument."
+)
+
+
+def jit(
+    func_or_sig=None,
+    device=False,
+    inline=False,
+    link=[],
+    debug=None,
+    opt=None,
+    lineinfo=False,
+    cache=False,
+    **kws,
+):
     """
     JIT compile a Python function for CUDA GPUs.
 
@@ -55,39 +66,43 @@ def jit(func_or_sig=None, device=False, inline=False, link=[], debug=None,
     """
 
     if link and config.ENABLE_CUDASIM:
-        raise NotImplementedError('Cannot link PTX in the simulator')
+        raise NotImplementedError("Cannot link PTX in the simulator")
 
-    if kws.get('boundscheck'):
+    if kws.get("boundscheck"):
         raise NotImplementedError("bounds checking is not supported for CUDA")
 
-    if kws.get('argtypes') is not None:
-        msg = _msg_deprecated_signature_arg.format('argtypes')
+    if kws.get("argtypes") is not None:
+        msg = _msg_deprecated_signature_arg.format("argtypes")
         raise DeprecationError(msg)
-    if kws.get('restype') is not None:
-        msg = _msg_deprecated_signature_arg.format('restype')
+    if kws.get("restype") is not None:
+        msg = _msg_deprecated_signature_arg.format("restype")
         raise DeprecationError(msg)
-    if kws.get('bind') is not None:
-        msg = _msg_deprecated_signature_arg.format('bind')
+    if kws.get("bind") is not None:
+        msg = _msg_deprecated_signature_arg.format("bind")
         raise DeprecationError(msg)
 
     debug = config.CUDA_DEBUGINFO_DEFAULT if debug is None else debug
     opt = (config.OPT != 0) if opt is None else opt
-    fastmath = kws.get('fastmath', False)
-    extensions = kws.get('extensions', [])
+    fastmath = kws.get("fastmath", False)
+    extensions = kws.get("extensions", [])
 
     if debug and opt:
-        msg = ("debug=True with opt=True "
-               "is not supported by CUDA. This may result in a crash"
-               " - set debug=False or opt=False.")
+        msg = (
+            "debug=True with opt=True "
+            "is not supported by CUDA. This may result in a crash"
+            " - set debug=False or opt=False."
+        )
         warn(NumbaInvalidConfigWarning(msg))
 
     if debug and lineinfo:
-        msg = ("debug and lineinfo are mutually exclusive. Use debug to get "
-               "full debug info (this disables some optimizations), or "
-               "lineinfo for line info only with code generation unaffected.")
+        msg = (
+            "debug and lineinfo are mutually exclusive. Use debug to get "
+            "full debug info (this disables some optimizations), or "
+            "lineinfo for line info only with code generation unaffected."
+        )
         warn(NumbaInvalidConfigWarning(msg))
 
-    if device and kws.get('link'):
+    if device and kws.get("link"):
         raise ValueError("link keyword invalid for device function")
 
     if sigutils.is_signature(func_or_sig):
@@ -101,19 +116,21 @@ def jit(func_or_sig=None, device=False, inline=False, link=[], debug=None,
 
     if signatures is not None:
         if config.ENABLE_CUDASIM:
+
             def jitwrapper(func):
                 return FakeCUDAKernel(func, device=device, fastmath=fastmath)
+
             return jitwrapper
 
         def _jit(func):
             targetoptions = kws.copy()
-            targetoptions['debug'] = debug
-            targetoptions['lineinfo'] = lineinfo
-            targetoptions['link'] = link
-            targetoptions['opt'] = opt
-            targetoptions['fastmath'] = fastmath
-            targetoptions['device'] = device
-            targetoptions['extensions'] = extensions
+            targetoptions["debug"] = debug
+            targetoptions["lineinfo"] = lineinfo
+            targetoptions["link"] = link
+            targetoptions["opt"] = opt
+            targetoptions["fastmath"] = fastmath
+            targetoptions["device"] = device
+            targetoptions["extensions"] = extensions
 
             disp = CUDADispatcher(func, targetoptions=targetoptions)
 
@@ -128,6 +145,7 @@ def jit(func_or_sig=None, device=False, inline=False, link=[], debug=None,
 
                 if device:
                     from numba.core import typeinfer
+
                     with typeinfer.register_dispatcher(disp):
                         disp.compile_device(argtypes, restype)
                 else:
@@ -142,29 +160,41 @@ def jit(func_or_sig=None, device=False, inline=False, link=[], debug=None,
     else:
         if func_or_sig is None:
             if config.ENABLE_CUDASIM:
+
                 def autojitwrapper(func):
-                    return FakeCUDAKernel(func, device=device,
-                                          fastmath=fastmath)
+                    return FakeCUDAKernel(
+                        func, device=device, fastmath=fastmath
+                    )
             else:
+
                 def autojitwrapper(func):
-                    return jit(func, device=device, debug=debug, opt=opt,
-                               lineinfo=lineinfo, link=link, cache=cache, **kws)
+                    return jit(
+                        func,
+                        device=device,
+                        debug=debug,
+                        opt=opt,
+                        lineinfo=lineinfo,
+                        link=link,
+                        cache=cache,
+                        **kws,
+                    )
 
             return autojitwrapper
         # func_or_sig is a function
         else:
             if config.ENABLE_CUDASIM:
-                return FakeCUDAKernel(func_or_sig, device=device,
-                                      fastmath=fastmath)
+                return FakeCUDAKernel(
+                    func_or_sig, device=device, fastmath=fastmath
+                )
             else:
                 targetoptions = kws.copy()
-                targetoptions['debug'] = debug
-                targetoptions['lineinfo'] = lineinfo
-                targetoptions['opt'] = opt
-                targetoptions['link'] = link
-                targetoptions['fastmath'] = fastmath
-                targetoptions['device'] = device
-                targetoptions['extensions'] = extensions
+                targetoptions["debug"] = debug
+                targetoptions["lineinfo"] = lineinfo
+                targetoptions["opt"] = opt
+                targetoptions["link"] = link
+                targetoptions["fastmath"] = fastmath
+                targetoptions["device"] = device
+                targetoptions["extensions"] = extensions
                 disp = CUDADispatcher(func_or_sig, targetoptions=targetoptions)
 
                 if cache:
@@ -191,7 +221,7 @@ def declare_device(name, sig, link=None):
 
     argtypes, restype = sigutils.normalize_signature(sig)
     if restype is None:
-        msg = 'Return type must be provided for device declarations'
+        msg = "Return type must be provided for device declarations"
         raise TypeError(msg)
 
     return declare_device_function(name, restype, argtypes, link)

numba_cuda/numba/cuda/descriptor.py

@@ -30,4 +30,4 @@ class CUDATarget(TargetDescriptor):
         return self._targetctx
 
 
-cuda_target = CUDATarget('cuda')
+cuda_target = CUDATarget("cuda")

numba_cuda/numba/cuda/device_init.py

@@ -1,21 +1,58 @@
 # Re export
 import sys
 from numba.cuda import cg
-from .stubs import (threadIdx, blockIdx, blockDim, gridDim, laneid, warpsize,
-                    syncwarp, shared, local, const, atomic,
-                    shfl_sync_intrinsic, vote_sync_intrinsic, match_any_sync,
-                    match_all_sync, threadfence_block, threadfence_system,
-                    threadfence, selp, popc, brev, clz, ffs, fma, cbrt,
-                    activemask, lanemask_lt, nanosleep, fp16,
-                    _vector_type_stubs)
-from .intrinsics import (grid, gridsize, syncthreads, syncthreads_and,
-                         syncthreads_count, syncthreads_or)
+from .stubs import (
+    threadIdx,
+    blockIdx,
+    blockDim,
+    gridDim,
+    laneid,
+    warpsize,
+    syncwarp,
+    shared,
+    local,
+    const,
+    atomic,
+    shfl_sync_intrinsic,
+    vote_sync_intrinsic,
+    match_any_sync,
+    match_all_sync,
+    threadfence_block,
+    threadfence_system,
+    threadfence,
+    selp,
+    popc,
+    brev,
+    clz,
+    ffs,
+    fma,
+    cbrt,
+    activemask,
+    lanemask_lt,
+    nanosleep,
+    fp16,
+    _vector_type_stubs,
+)
+from .intrinsics import (
+    grid,
+    gridsize,
+    syncthreads,
+    syncthreads_and,
+    syncthreads_count,
+    syncthreads_or,
+)
 from .cudadrv.error import CudaSupportError
-from numba.cuda.cudadrv.driver import (BaseCUDAMemoryManager,
-                                       HostOnlyCUDAMemoryManager,
-                                       GetIpcHandleMixin, MemoryPointer,
-                                       MappedMemory, PinnedMemory, MemoryInfo,
-                                       IpcHandle, set_memory_manager)
+from numba.cuda.cudadrv.driver import (
+    BaseCUDAMemoryManager,
+    HostOnlyCUDAMemoryManager,
+    GetIpcHandleMixin,
+    MemoryPointer,
+    MappedMemory,
+    PinnedMemory,
+    MemoryInfo,
+    IpcHandle,
+    set_memory_manager,
+)
 from numba.cuda.cudadrv.runtime import runtime
 from .cudadrv import nvvm
 from numba.cuda import initialize
@@ -26,13 +63,27 @@ from .api import *
 from .api import _auto_device
 from .args import In, Out, InOut
 
-from .intrinsic_wrapper import (all_sync, any_sync, eq_sync, ballot_sync,
-                                shfl_sync, shfl_up_sync, shfl_down_sync,
-                                shfl_xor_sync)
+from .intrinsic_wrapper import (
+    all_sync,
+    any_sync,
+    eq_sync,
+    ballot_sync,
+    shfl_sync,
+    shfl_up_sync,
+    shfl_down_sync,
+    shfl_xor_sync,
+)
 
 from .kernels import reduction
 from numba.cuda.cudadrv.linkable_code import (
-    Archive, CUSource, Cubin, Fatbin, LinkableCode, LTOIR, Object, PTXSource
+    Archive,
+    CUSource,
+    Cubin,
+    Fatbin,
+    LinkableCode,
+    LTOIR,
+    Object,
+    PTXSource,
 )
 
 reduce = Reduce = reduction.Reduce