Skip to content
This repository was archived by the owner on Jan 25, 2023. It is now read-only.

Commit cdf7377

Browse files
author
etotmeni
committed
Del ocldrv
1 parent e31c726 commit cdf7377

File tree

3 files changed

+2
-5
lines changed

3 files changed

+2
-5
lines changed

numba/dppl/examples/dppl_with_context.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
import numpy as np
22
from numba import dppl, njit, prange
33
import dpctl
4-
import dpctl.ocldrv as ocldrv
54

65

76
@njit
@@ -21,10 +20,10 @@ def main():
2120
b = np.ones(N)
2221
c = np.ones(N)
2322

24-
if ocldrv.has_gpu_device:
23+
if dpctl.has_gpu_queues():
2524
with dpctl.device_context(dpctl.device_type.gpu):
2625
f(a, b, c, N)
27-
elif ocldrv.has_cpu_device:
26+
elif dpctl.has_cpu_queues():
2827
with dpctl.device_context(dpctl.device_type.cpu):
2928
f(a, b, c, N)
3029
else:

numba/dppl/target_dispatcher.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22
from numba import types
33
from numba.core.errors import UnsupportedError
44
import dpctl
5-
import dpctl.ocldrv as ocldrv
65
from numba.core.compiler_lock import global_compiler_lock
76

87

numba/dppl/tests/dppl/test_with_context.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55
from numba.tests.support import captured_stdout
66
from numba.dppl.testing import DPPLTestCase, unittest
77
import dpctl
8-
import dpctl.ocldrv as ocldrv
98

109

1110
@unittest.skipIf(not dpctl.has_gpu_queues(), "No GPU platforms available")

0 commit comments

Comments
 (0)