numba-cuda 0.0.21__py3-none-any.whl → 0.2.0__py3-none-any.whl

_numba_cuda_redirector.py

@@ -67,9 +67,15 @@ class NumbaCudaFinder(importlib.abc.MetaPathFinder):
             oot_path = [p.replace(self.numba_path, self.numba_cuda_path)
                         for p in path]
             for finder in sys.meta_path:
-                spec = finder.find_spec(name, oot_path, target)
-                if spec is not None:
-                    return spec
+                try:
+                    spec = finder.find_spec(name, oot_path, target)
+                except AttributeError:
+                    # Finders written to a pre-Python 3.4 spec for finders will
+                    # not implement find_spec. We can skip those altogether.
+                    continue
+                else:
+                    if spec is not None:
+                        return spec
 
 
 finder = NumbaCudaFinder()

numba_cuda/VERSION

@@ -1 +1 @@
-0.0.21
+0.2.0

numba_cuda/numba/cuda/compiler.py

@@ -206,6 +206,7 @@ def compile_cuda(pyfunc, return_type, args, debug=False, lineinfo=False,
 
     if debug:
         flags.error_model = 'python'
+        flags.dbg_extend_lifetimes = True
     else:
         flags.error_model = 'numpy'
 
@@ -497,6 +498,9 @@ def compile(pyfunc, sig, debug=None, lineinfo=False, device=True,
         'opt': 3 if opt else 0
     }
 
+    if debug:
+        nvvm_options['g'] = None
+
     if lto:
         nvvm_options['gen-lto'] = None
 

numba_cuda/numba/cuda/cudadecl.py

@@ -508,6 +508,11 @@ class Cuda_atomic_cas(AbstractTemplate):
             return signature(dty, ary, idx, dty, dty)
 
 
+@register_global(breakpoint)
+class Cuda_breakpoint(ConcreteTemplate):
+    cases = [signature(types.none)]
+
+
 @register
 class Cuda_nanosleep(ConcreteTemplate):
     key = cuda.nanosleep

numba_cuda/numba/cuda/cudadrv/driver.py

@@ -42,6 +42,11 @@ from .mappings import FILE_EXTENSION_MAP
 from .linkable_code import LinkableCode, LTOIR, Fatbin, Object
 from numba.cuda.cudadrv import enums, drvapi, nvrtc
 
+try:
+    from pynvjitlink.api import NvJitLinker, NvJitLinkError
+except ImportError:
+    NvJitLinker, NvJitLinkError = None, None
+
 USE_NV_BINDING = config.CUDA_USE_NVIDIA_BINDING
 
 if USE_NV_BINDING:
@@ -92,20 +97,6 @@ ENABLE_PYNVJITLINK = (
 if not hasattr(config, "CUDA_ENABLE_PYNVJITLINK"):
     config.CUDA_ENABLE_PYNVJITLINK = ENABLE_PYNVJITLINK
 
-if ENABLE_PYNVJITLINK:
-    try:
-        from pynvjitlink.api import NvJitLinker, NvJitLinkError
-    except ImportError:
-        raise ImportError(
-            "Using pynvjitlink requires the pynvjitlink package to be available"
-        )
-
-    if config.CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY:
-        raise ValueError(
-            "Can't set CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY and "
-            "CUDA_ENABLE_PYNVJITLINK at the same time"
-        )
-
 
 def make_logger():
     logger = logging.getLogger(__name__)
@@ -3061,6 +3052,17 @@ class PyNvJitLinker(Linker):
         lto=False,
         additional_flags=None,
     ):
+        if NvJitLinker is None:
+            raise ImportError(
+                "Using pynvjitlink requires the pynvjitlink package to be "
+                "available"
+            )
+
+        if config.CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY:
+            raise ValueError(
+                "Can't set CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY and "
+                "CUDA_ENABLE_PYNVJITLINK at the same time"
+            )
 
         if cc is None:
             raise RuntimeError("PyNvJitLinker requires CC to be specified")

numba_cuda/numba/cuda/cudaimpl.py

@@ -934,6 +934,14 @@ def ptx_atomic_cas(context, builder, sig, args):
 
 # -----------------------------------------------------------------------------
 
+
+@lower(breakpoint)
+def ptx_brkpt(context, builder, sig, args):
+    brkpt = ir.InlineAsm(ir.FunctionType(ir.VoidType(), []),
+                         "brkpt;", '', side_effect=True)
+    builder.call(brkpt, ())
+
+
 @lower(stubs.nanosleep, types.uint32)
 def ptx_nanosleep(context, builder, sig, args):
     nanosleep = ir.InlineAsm(ir.FunctionType(ir.VoidType(), [ir.IntType(32)]),

numba_cuda/numba/cuda/dispatcher.py

@@ -95,6 +95,9 @@ class _Kernel(serialize.ReduceMixin):
             'opt': 3 if opt else 0
         }
 
+        if debug:
+            nvvm_options['g'] = None
+
         cc = get_current_device().compute_capability
         cres = compile_cuda(self.py_func, types.void, self.argtypes,
                             debug=self.debug,
@@ -918,6 +921,9 @@ class CUDADispatcher(Dispatcher, serialize.ReduceMixin):
                     'fastmath': fastmath
                 }
 
+                if debug:
+                    nvvm_options['g'] = None
+
                 cc = get_current_device().compute_capability
                 cres = compile_cuda(self.py_func, return_type, args,
                                     debug=debug,

numba_cuda/numba/cuda/tests/cudadrv/test_nvjitlink.py

@@ -2,6 +2,18 @@ from numba.cuda.testing import unittest
 from numba.cuda.testing import skip_on_cudasim
 from numba.cuda.testing import CUDATestCase
 from numba.cuda.cudadrv.driver import PyNvJitLinker
+from numba.cuda import get_current_device
+
+from numba import cuda
+from numba import config
+from numba.tests.support import run_in_subprocess, override_config
+
+try:
+    import pynvjitlink  # noqa: F401
+    PYNVJITLINK_INSTALLED = True
+except ImportError:
+    PYNVJITLINK_INSTALLED = False
+
 
 import itertools
 import os
@@ -9,9 +21,6 @@ import io
 import contextlib
 import warnings
 
-from numba.cuda import get_current_device
-from numba import cuda
-from numba import config
 
 TEST_BIN_DIR = os.getenv("NUMBA_CUDA_TEST_BIN_DIR")
 if TEST_BIN_DIR:
@@ -251,5 +260,69 @@ class TestLinker(CUDATestCase):
                 pass
 
 
+@unittest.skipIf(
+    not PYNVJITLINK_INSTALLED, reason="Pynvjitlink is not installed"
+)
+class TestLinkerUsage(CUDATestCase):
+    """Test that whether pynvjitlink can be enabled by both environment variable
+    and modification of config at runtime.
+    """
+
+    src = """if 1:
+        import os
+        from numba import cuda, config
+
+        {config}
+
+        TEST_BIN_DIR = os.getenv("NUMBA_CUDA_TEST_BIN_DIR")
+        if TEST_BIN_DIR:
+            test_device_functions_cubin = os.path.join(
+                TEST_BIN_DIR, "test_device_functions.cubin"
+            )
+
+        sig = "uint32(uint32, uint32)"
+        add_from_numba = cuda.declare_device("add_from_numba", sig)
+
+        @cuda.jit(link=[test_device_functions_cubin], lto=True)
+        def kernel(result):
+            result[0] = add_from_numba(1, 2)
+
+        result = cuda.device_array(1)
+        kernel[1, 1](result)
+        assert result[0] == 3
+        """
+
+    def test_linker_enabled_envvar(self):
+        env = os.environ.copy()
+        env['NUMBA_CUDA_ENABLE_PYNVJITLINK'] = "1"
+        run_in_subprocess(self.src.format(config=""), env=env)
+
+    def test_linker_disabled_envvar(self):
+        env = os.environ.copy()
+        env.pop('NUMBA_CUDA_ENABLE_PYNVJITLINK', None)
+        with self.assertRaisesRegex(
+            AssertionError, "LTO and additional flags require PyNvJitLinker"
+        ):
+            # Actual error raised is `ValueError`, but `run_in_subprocess`
+            # reraises as AssertionError.
+            run_in_subprocess(self.src.format(config=""), env=env)
+
+    def test_linker_enabled_config(self):
+        env = os.environ.copy()
+        env.pop('NUMBA_CUDA_ENABLE_PYNVJITLINK', None)
+        run_in_subprocess(self.src.format(
+            config="config.CUDA_ENABLE_PYNVJITLINK = True"), env=env)
+
+    def test_linker_disabled_config(self):
+        env = os.environ.copy()
+        env.pop('NUMBA_CUDA_ENABLE_PYNVJITLINK', None)
+        with override_config("CUDA_ENABLE_PYNVJITLINK", False):
+            with self.assertRaisesRegex(
+                AssertionError, "LTO and additional flags require PyNvJitLinker"
+            ):
+                run_in_subprocess(self.src.format(
+                    config="config.CUDA_ENABLE_PYNVJITLINK = False"), env=env)
+
+
 if __name__ == "__main__":
     unittest.main()

{numba_cuda-0.0.21.dist-info → numba_cuda-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: numba-cuda
-Version: 0.0.21
+Version: 0.2.0
 Summary: CUDA target for Numba
 Author: Anaconda Inc., NVIDIA Corporation
 License: BSD 2-clause

{numba_cuda-0.0.21.dist-info → numba_cuda-0.2.0.dist-info}/RECORD

@@ -1,6 +1,6 @@
 _numba_cuda_redirector.pth,sha256=cmfMMmV0JPh3yEpl4bGeM9AuXiVVMSo6Z_b7RaQL3XE,30
-_numba_cuda_redirector.py,sha256=rc56rnb40w3AtrqnhS66JSgYTSTsi3iTn8yP3NuoQV8,2401
-numba_cuda/VERSION,sha256=N0wu4MReU0U_7uoeU-17rOqTT3ZYtrLE_x8SJjefmc8,7
+_numba_cuda_redirector.py,sha256=QKJmYICSQvjvph0Zw9OW015MsuKxIF28GPFjR35AXLM,2681
+numba_cuda/VERSION,sha256=H5MN0fEzwfl6lP46y42zQ3LPTAH_2ys_9Mpy-UlBIek,6
 numba_cuda/__init__.py,sha256=atXeUvJKR3JHcAiCFbXCVOJQUHgB1TulmsqSL_9RT3Q,114
 numba_cuda/_version.py,sha256=jbdUsbR7sVllw0KxQNB0-FMd929CGg3kH2fhHdrlkuc,719
 numba_cuda/numba/cuda/__init__.py,sha256=idyVHOObC9lTYnp62v7rVprSacRM4d5F6vhXfG5ElTI,621
@@ -9,19 +9,19 @@ numba_cuda/numba/cuda/api_util.py,sha256=aQfUV2-4RM_oGVvckMjbMr5e3effOQNX04v1T0O
 numba_cuda/numba/cuda/args.py,sha256=HloHkw_PQal2DT-I70Xf_XbnGObS1jiUgcRrQ85Gq28,1978
 numba_cuda/numba/cuda/cg.py,sha256=9V1uZqyGOJX1aFd9c6GAPbLSqq83lE8LoP-vxxrKENY,1490
 numba_cuda/numba/cuda/codegen.py,sha256=ghdYBKZ3Mzk2UlLE64HkrAjb60PN9fibSNkWFRQuj4M,13184
-numba_cuda/numba/cuda/compiler.py,sha256=XQHzUCuXl6WCtWWxv1X3Y9ebcVQVJEkzOuckNwKa4Gg,21249
+numba_cuda/numba/cuda/compiler.py,sha256=_0qfSjnLnF29B-t8NQRJt4FBUIKxZJE6xN47_G7oRio,21339
 numba_cuda/numba/cuda/cpp_function_wrappers.cu,sha256=iv84_F6Q9kFjV_kclrQz1msh6Dud8mI3qNkswTid7Qc,953
 numba_cuda/numba/cuda/cuda_fp16.h,sha256=1IC0mdNdkvKbvAe0-f4uYVS7WFrVqOyI1nRUbBiqr6A,126844
 numba_cuda/numba/cuda/cuda_fp16.hpp,sha256=vJ7NUr2X2tKhAP7ojydAiCoOjVO6n4QGoXD6m9Srrlw,89130
 numba_cuda/numba/cuda/cuda_paths.py,sha256=C0gA72QLWUMfvXkFpw1WqqaFqfsQ7HM72hQVXG0A7RU,10023
-numba_cuda/numba/cuda/cudadecl.py,sha256=ynUidit8oPGjedc6p1miMGtS20DOji3DiQHzwmx6m0s,23192
-numba_cuda/numba/cuda/cudaimpl.py,sha256=3YMxQSCv2KClBrpuXGchrTNICV1F6NIjjL2rie5fDZ4,38628
+numba_cuda/numba/cuda/cudadecl.py,sha256=ZUssRdTvS4sVwvJWTmaRTvrMXMbkPZ_qVp8JMXoXFoc,23300
+numba_cuda/numba/cuda/cudaimpl.py,sha256=0oHjDwBC4JmfpwS1Fsn1bm5YWVru5vZvvnO414P4TS0,38840
 numba_cuda/numba/cuda/cudamath.py,sha256=EFNtdzEytAZuwijdRoFGzVKCeal76UzzaNy7wUFQx8I,3978
 numba_cuda/numba/cuda/decorators.py,sha256=qSpir16-jPYSe2YuRZ6g9INeobmsMNg6ab9IZpwJocM,7823
 numba_cuda/numba/cuda/descriptor.py,sha256=rNMaurJkjNjIBmHPozDoLC35DMURE0fn_LtnXRmaG_w,985
 numba_cuda/numba/cuda/device_init.py,sha256=lP79tCsQ0Np9xcbjv_lXcH4JOiVZvV8nwg3INdETxsc,3586
 numba_cuda/numba/cuda/deviceufunc.py,sha256=yxAH71dpgJWK8okmCJm0FUV6z2AqdThCYOTZspT7z0M,30775
-numba_cuda/numba/cuda/dispatcher.py,sha256=JuUr0-6xQtDkyaZv7CirWaU5_sSNX4BKCTDgQG5c1xc,41116
+numba_cuda/numba/cuda/dispatcher.py,sha256=nDfPCzxJ7UwA4uiz-fsMMgQb2WXByvzHLtkLMXW9JXk,41244
 numba_cuda/numba/cuda/errors.py,sha256=XwWHzCllx0DXU6BQdoRH0m3pznGxnTFOBTVYXMmCfqg,1724
 numba_cuda/numba/cuda/extending.py,sha256=URsyBYls2te-mgE0yvDY6akvawYCA0blBFfD7Lf9DO4,142
 numba_cuda/numba/cuda/initialize.py,sha256=TQGHGLQoq4ch4J6CLDcJdGsZzXM-g2kDgdyO1u-Rbhg,546
@@ -47,7 +47,7 @@ numba_cuda/numba/cuda/vectorizers.py,sha256=u_0EzaD5tqVH8uOz4Gmqn3FgPC1rckwDAQuR
 numba_cuda/numba/cuda/cudadrv/__init__.py,sha256=0TL4MZcJXUoo9qA7uu0vLv7eHrXRerVmyfi7O149ITw,199
 numba_cuda/numba/cuda/cudadrv/devicearray.py,sha256=06kM7iFcx1TYiFhs1o9r1kyoA3k5yS7mFAdZDf6nrxA,31215
 numba_cuda/numba/cuda/cudadrv/devices.py,sha256=6SneNmoq83gue0txFWWx4A65vViAa8xA06FzkApoqAk,7992
-numba_cuda/numba/cuda/cudadrv/driver.py,sha256=bjlGcJvyjwMjRCNkNqmBIAA0HO_fzbrW2afXsp-YiCg,114794
+numba_cuda/numba/cuda/cudadrv/driver.py,sha256=FONYaUzgexmPUIMsSq0zr_FgD9eLbWT8m1APEVrLJRo,114887
 numba_cuda/numba/cuda/cudadrv/drvapi.py,sha256=52ms3X6hfPaQB8E1jb6g7QKqRvHzBMlDQ-V2DM1rXxQ,17178
 numba_cuda/numba/cuda/cudadrv/dummyarray.py,sha256=nXRngdr-k3h_BNGQuJUxmp89yGNWxqEDJedpwDPEZ44,14209
 numba_cuda/numba/cuda/cudadrv/enums.py,sha256=Wy5dzukTk4TnWCowg_PLceET_v2xEyiWLu9TyH8pXr8,23742
@@ -103,7 +103,7 @@ numba_cuda/numba/cuda/tests/cudadrv/test_is_fp16.py,sha256=0KPe4E9wOZsSV_0QI0Lmj
 numba_cuda/numba/cuda/tests/cudadrv/test_linker.py,sha256=_l2_EQEko2Jet5ooj4XMT0L4BjOuqLjbONGj1_MVI50,10161
 numba_cuda/numba/cuda/tests/cudadrv/test_managed_alloc.py,sha256=kYXYMkx_3GPAITKp4reLeM8KSzKkpxiC8nxnBvXpaTA,4979
 numba_cuda/numba/cuda/tests/cudadrv/test_mvc.py,sha256=984jATSa01SRoSrVqxPeO6ujJ7w2jsnZa39ABInFLVI,1529
-numba_cuda/numba/cuda/tests/cudadrv/test_nvjitlink.py,sha256=VOOl5fLxQL5IKHEi8hL47hAH0BUf_D8NyIxptLxIwus,8856
+numba_cuda/numba/cuda/tests/cudadrv/test_nvjitlink.py,sha256=oZywLDuX-l1LJvIIU4QCsE7UCwtIKbBP7u6GxVDpD_g,11316
 numba_cuda/numba/cuda/tests/cudadrv/test_nvvm_driver.py,sha256=DF7KV5uh-yMztks0f47NhpalV64dvsNy-f8HY6GhAhE,7373
 numba_cuda/numba/cuda/tests/cudadrv/test_pinned.py,sha256=u_TthSS2N-2J4eBIuF4PGg33AjD-wxly7MKpz0vRAKc,944
 numba_cuda/numba/cuda/tests/cudadrv/test_profiler.py,sha256=MQWZx1j3lbEpWmIpQ1bV9szrGOV3VHN0QrEnJRjAhW4,508
@@ -236,8 +236,8 @@ numba_cuda/numba/cuda/tests/test_binary_generation/Makefile,sha256=P2WzCc5d64JGq
 numba_cuda/numba/cuda/tests/test_binary_generation/generate_raw_ltoir.py,sha256=V0raLZLGSiWbE_K-JluI0CnmNkXbhlMVj-TH7P1OV8E,5014
 numba_cuda/numba/cuda/tests/test_binary_generation/test_device_functions.cu,sha256=cUf-t6ZM9MK_x7X_aKwsrKW1LdR97XcpR-qnYr5faOE,453
 numba_cuda/numba/cuda/tests/test_binary_generation/undefined_extern.cu,sha256=q3oxZziT8KDodeNcEBiWULH6vMrHCWucmJmtrg8C0d0,128
-numba_cuda-0.0.21.dist-info/LICENSE,sha256=eHeYE-XjASmwbxfsP5AImgfzRwZurZGqH1f6OFwJ4io,1326
-numba_cuda-0.0.21.dist-info/METADATA,sha256=U_oWdBsw_mdsI2AnFJDXdxTXL2ytOeuTHwS3wCZswTI,1497
-numba_cuda-0.0.21.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-numba_cuda-0.0.21.dist-info/top_level.txt,sha256=C50SsH-8tXDmt7I0Y3nlJYhS5s6pqWflCPdobe9vx2M,11
-numba_cuda-0.0.21.dist-info/RECORD,,
+numba_cuda-0.2.0.dist-info/LICENSE,sha256=eHeYE-XjASmwbxfsP5AImgfzRwZurZGqH1f6OFwJ4io,1326
+numba_cuda-0.2.0.dist-info/METADATA,sha256=u3e2Hm6iPkdyyDwsvGJ7B3RecpE7X3zA2SHrX-z7Kc4,1496
+numba_cuda-0.2.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+numba_cuda-0.2.0.dist-info/top_level.txt,sha256=C50SsH-8tXDmt7I0Y3nlJYhS5s6pqWflCPdobe9vx2M,11
+numba_cuda-0.2.0.dist-info/RECORD,,