Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions numba_dppy/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -506,14 +506,14 @@ def main():
from numba import config
import numba.testing

from numba.dppl_config import *
if dppl_present:
from .config import dppy_present
if dppy_present:
from .device_init import *
else:
raise ImportError("Importing dppl failed")

def test(*args, **kwargs):
if not dppl_present and not is_available():
if not dppy_present and not is_available():
dppl_error()

return numba.testing.test("numba_dppy.tests", *args, **kwargs)
Expand Down
6 changes: 6 additions & 0 deletions numba_dppy/config.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,6 @@
try:
import dpctl

dppy_present = dpctl.has_sycl_platforms() and dpctl.has_gpu_queues()
except:
dppy_present = False
4 changes: 2 additions & 2 deletions numba_dppy/dppl_offload_dispatcher.py
Original file line number Diff line number Diff line change
@@ -1,13 +1,13 @@
from numba.core import dispatcher, compiler
from numba.core.registry import cpu_target, dispatcher_registry
import numba.dppl_config as dppl_config
import numba_dppy.config as dppy_config


class DpplOffloadDispatcher(dispatcher.Dispatcher):
targetdescr = cpu_target

def __init__(self, py_func, locals={}, targetoptions={}, impl_kind='direct', pipeline_class=compiler.Compiler):
if dppl_config.dppl_present:
if dppy_config.dppy_present:
from numba_dppy.compiler import DPPLCompiler
targetoptions['parallel'] = True
dispatcher.Dispatcher.__init__(self, py_func, locals=locals,
Expand Down
4 changes: 2 additions & 2 deletions numba_dppy/tests/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,14 +3,14 @@
from os.path import dirname, join


import numba.dppl_config as dppl_config
import numba_dppy.config as dppy_config

def load_tests(loader, tests, pattern):

suite = SerialSuite()
this_dir = dirname(__file__)

if dppl_config.dppl_present:
if dppy_config.dppy_present:
suite.addTests(load_testsuite(loader, join(this_dir, 'dppl')))
else:
print("skipped DPPL tests")
Expand Down