4704264818

Build Type

github

Committed by GitHub

Commit Message

Merge adb71d616 into f9a77a2c3

Pull Request Pull Request #1001: Feature/add numba rc support

Run Details

1906 of 3555 branches covered (53.61%)

Branch coverage included in aggregate %.

2 of 3 new or added lines in 1 file covered. (66.67%)

4 existing lines in 1 file now uncovered.

6700 of 10176 relevant lines covered (65.84%)

0.66 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

85.51

/numba_dpex/__init__.py

# SPDX-FileCopyrightText: 2020 - 2023 Intel Corporation
#
# SPDX-License-Identifier: Apache-2.0

"""
The numba-dpex extension module adds data-parallel offload support to Numba.
"""
import glob
import logging
import os
import platform as plt
import re

import dpctl
import llvmlite.binding as ll
import numba
from numba.core import ir_utils
from numba.np import arrayobj
from numba.np.ufunc import array_exprs
from numba.np.ufunc.decorators import Vectorize

from numba_dpex._patches import _empty_nd_impl, _is_ufunc, _mk_alloc
from numba_dpex.vectorizers import Vectorize as DpexVectorize

# Monkey patches
array_exprs._is_ufunc = _is_ufunc
ir_utils.mk_alloc = _mk_alloc
arrayobj._empty_nd_impl = _empty_nd_impl


def load_dpctl_sycl_interface():
    """Permanently loads the ``DPCTLSyclInterface`` library provided by dpctl.
    The ``DPCTLSyclInterface`` library provides C wrappers over SYCL functions
    that are directly invoked from the LLVM modules generated by numba_dpex.
    We load the library once at the time of initialization using llvmlite's
    load_library_permanently function.
    Raises:
        ImportError: If the ``DPCTLSyclInterface`` library could not be loaded.
    """

    platform = plt.system()
    if platform == "Windows":
        paths = glob.glob(
            os.path.join(
                os.path.dirname(dpctl.__file__), "*DPCTLSyclInterface.dll"
            )
        )
    else:
        paths = glob.glob(
            os.path.join(
                os.path.dirname(dpctl.__file__), "*DPCTLSyclInterface.so.0"
            )
        )

    if len(paths) == 1:
        ll.load_library_permanently(paths[0])
    else:
        raise ImportError

    Vectorize.target_registry.ondemand["dpex"] = lambda: DpexVectorize


numba_version = tuple(
    map(
        int,
        re.sub(
            "([0-9]+\\.[0-9]+\\.[0-9]+).*",
            "\\g<1>",
            numba.__version__,
        ).split(".")[:3],
    )
)
if numba_version < (0, 56, 4):
    logging.warning(
        "numba_dpex needs numba 0.56.4, using "
        f"numba={numba_version} may cause unexpected behavior"
    )


dpctl_version = tuple(map(int, dpctl.__version__.split(".")[:2]))
if dpctl_version < (0, 14):
    logging.warning(
        "numba_dpex needs dpctl 0.14 or greater, using "
        f"dpctl={dpctl_version} may cause unexpected behavior"
    )

from numba import prange  # noqa E402

import numba_dpex.core.dpjit_dispatcher  # noqa E402
import numba_dpex.core.offload_dispatcher  # noqa E402

# Initialize the _dpexrt_python extension
import numba_dpex.core.runtime  # noqa E402
import numba_dpex.core.targets.dpjit_target  # noqa E402

# Re-export types itself
import numba_dpex.core.types as types  # noqa E402
from numba_dpex import config  # noqa E402
from numba_dpex.core.kernel_interface.indexers import (  # noqa E402
    NdRange,
    Range,
)

# Re-export all type names
from numba_dpex.core.types import *  # noqa E402
from numba_dpex.dpnp_iface import dpnpimpl  # noqa E402
from numba_dpex.retarget import offload_to_sycl_device  # noqa E402

if config.HAS_NON_HOST_DEVICE:
    # Re export
    from .core.targets import dpjit_target, kernel_target
    from .decorators import dpjit, func, kernel

    # We are importing dpnp stub module to make Numba recognize the
    # module when we rename Numpy functions.
    from .dpnp_iface.stubs import dpnp
    from .ocl.stubs import (
        GLOBAL_MEM_FENCE,
        LOCAL_MEM_FENCE,
        atomic,
        barrier,
        get_global_id,
        get_global_size,
        get_group_id,
        get_local_id,
        get_local_size,
        get_num_groups,
        get_work_dim,
        local,
        mem_fence,
        private,
        sub_group_barrier,
    )

    DEFAULT_LOCAL_SIZE = []
    load_dpctl_sycl_interface()
    del load_dpctl_sycl_interface
else:
    raise ImportError("No non-host SYCL device found to execute kernels.")

from numba_dpex._version import get_versions  # noqa E402

__version__ = get_versions()["version"]
del get_versions

__all__ = types.__all__ + ["offload_to_sycl_device"] + ["Range", "NdRange"]

1	# SPDX-FileCopyrightText: 2020 - 2023 Intel Corporation
2	#
3	# SPDX-License-Identifier: Apache-2.0
4
5	"""	1✔
6	The numba-dpex extension module adds data-parallel offload support to Numba.
7	"""
8	import glob	1✔
9	import logging	1✔
10	import os	1✔
11	import platform as plt	1✔
12	import re	1✔
13
14	import dpctl	1✔
15	import llvmlite.binding as ll	1✔
16	import numba	1✔
17	from numba.core import ir_utils	1✔
18	from numba.np import arrayobj	1✔
19	from numba.np.ufunc import array_exprs	1✔
20	from numba.np.ufunc.decorators import Vectorize	1✔
21
22	from numba_dpex._patches import _empty_nd_impl, _is_ufunc, _mk_alloc	1✔
23	from numba_dpex.vectorizers import Vectorize as DpexVectorize	1✔
24
25	# Monkey patches
26	array_exprs._is_ufunc = _is_ufunc	1✔
27	ir_utils.mk_alloc = _mk_alloc	1✔
28	arrayobj._empty_nd_impl = _empty_nd_impl	1✔
29
30
31	def load_dpctl_sycl_interface():	1✔
32	"""Permanently loads the ``DPCTLSyclInterface`` library provided by dpctl.
33	The ``DPCTLSyclInterface`` library provides C wrappers over SYCL functions
34	that are directly invoked from the LLVM modules generated by numba_dpex.
35	We load the library once at the time of initialization using llvmlite's
36	load_library_permanently function.
37	Raises:
38	ImportError: If the ``DPCTLSyclInterface`` library could not be loaded.
39	"""
40
41	platform = plt.system()	1✔
42	if platform == "Windows":	1!
UNCOV 43	paths = glob.glob(	×
44	os.path.join(
45	os.path.dirname(dpctl.__file__), "*DPCTLSyclInterface.dll"
46	)
47	)
48	else:
49	paths = glob.glob(	1✔
50	os.path.join(
51	os.path.dirname(dpctl.__file__), "*DPCTLSyclInterface.so.0"
52	)
53	)
54
55	if len(paths) == 1:	1!
56	ll.load_library_permanently(paths[0])	1✔
57	else:
UNCOV 58	raise ImportError	×
59
60	Vectorize.target_registry.ondemand["dpex"] = lambda: DpexVectorize	1✔
61
62
63	numba_version = tuple(	1✔
64	map(
65	int,
66	re.sub(
67	"([0-9]+\\.[0-9]+\\.[0-9]+).*",
68	"\\g<1>",
69	numba.__version__,
70	).split(".")[:3],
71	)
72	)
73	if numba_version < (0, 56, 4):	1!
NEW 74	logging.warning(	×
75	"numba_dpex needs numba 0.56.4, using "
76	f"numba={numba_version} may cause unexpected behavior"
77	)
78
79
80	dpctl_version = tuple(map(int, dpctl.__version__.split(".")[:2]))	1✔
81	if dpctl_version < (0, 14):	1!
UNCOV 82	logging.warning(	×
83	"numba_dpex needs dpctl 0.14 or greater, using "
84	f"dpctl={dpctl_version} may cause unexpected behavior"
85	)
86
87	from numba import prange # noqa E402	1✔
88
89	import numba_dpex.core.dpjit_dispatcher # noqa E402	1✔
90	import numba_dpex.core.offload_dispatcher # noqa E402	1✔
91
92	# Initialize the _dpexrt_python extension
93	import numba_dpex.core.runtime # noqa E402	1✔
94	import numba_dpex.core.targets.dpjit_target # noqa E402	1✔
95
96	# Re-export types itself
97	import numba_dpex.core.types as types # noqa E402	1✔
98	from numba_dpex import config # noqa E402	1✔
99	from numba_dpex.core.kernel_interface.indexers import ( # noqa E402	1✔
100	NdRange,
101	Range,
102	)
103
104	# Re-export all type names
105	from numba_dpex.core.types import * # noqa E402	1✔
106	from numba_dpex.dpnp_iface import dpnpimpl # noqa E402	1✔
107	from numba_dpex.retarget import offload_to_sycl_device # noqa E402	1✔
108
109	if config.HAS_NON_HOST_DEVICE:	1!
110	# Re export
111	from .core.targets import dpjit_target, kernel_target	1✔
112	from .decorators import dpjit, func, kernel	1✔
113
114	# We are importing dpnp stub module to make Numba recognize the
115	# module when we rename Numpy functions.
116	from .dpnp_iface.stubs import dpnp	1✔
117	from .ocl.stubs import (	1✔
118	GLOBAL_MEM_FENCE,
119	LOCAL_MEM_FENCE,
120	atomic,
121	barrier,
122	get_global_id,
123	get_global_size,
124	get_group_id,
125	get_local_id,
126	get_local_size,
127	get_num_groups,
128	get_work_dim,
129	local,
130	mem_fence,
131	private,
132	sub_group_barrier,
133	)
134
135	DEFAULT_LOCAL_SIZE = []	1✔
136	load_dpctl_sycl_interface()	1✔
137	del load_dpctl_sycl_interface	1✔
138	else:
UNCOV 139	raise ImportError("No non-host SYCL device found to execute kernels.")	×
140
141	from numba_dpex._version import get_versions # noqa E402	1✔
142
143	__version__ = get_versions()["version"]	1✔
144	del get_versions	1✔
145
146	__all__ = types.__all__ + ["offload_to_sycl_device"] + ["Range", "NdRange"]	1✔

IntelPython / numba-dpex / 4704264818

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous