File tree Expand file tree Collapse file tree 1 file changed +5
-6
lines changed Expand file tree Collapse file tree 1 file changed +5
-6
lines changed Original file line number Diff line number Diff line change 14
14
# See the License for the specific language governing permissions and
15
15
# limitations under the License.
16
16
17
- import dpctl
18
-
17
+ from ._sycl_device import SyclDevice
19
18
from ._usmarray import DLDeviceType
20
19
21
20
@@ -29,12 +28,12 @@ def dldevice_to_sycldevice(dl_dev: tuple):
29
28
)
30
29
if dl_dev [0 ] != DLDeviceType .kDLOneAPI :
31
30
raise ValueError ("dldevice type must be kDLOneAPI" )
32
- return dpctl . SyclDevice (str (dl_dev [1 ]))
31
+ return SyclDevice (str (dl_dev [1 ]))
33
32
34
33
35
- def sycldevice_to_dldevice (dev : dpctl . SyclDevice ):
36
- if not isinstance (dev , dpctl . SyclDevice ):
34
+ def sycldevice_to_dldevice (dev : SyclDevice ):
35
+ if not isinstance (dev , SyclDevice ):
37
36
raise TypeError (
38
- "dev is expected to be a dpctl. SyclDevice, got " f"{ type (dev )} "
37
+ "dev is expected to be a SyclDevice, got " f"{ type (dev )} "
39
38
)
40
39
return (DLDeviceType .kDLOneAPI , dev .get_device_id ())
You can’t perform that action at this time.
0 commit comments