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 1414# See the License for the specific language governing permissions and
1515# limitations under the License.
1616
17- import dpctl
18-
17+ from .._sycl_device import SyclDevice
1918from ._usmarray import DLDeviceType
2019
2120
@@ -29,12 +28,12 @@ def dldevice_to_sycldevice(dl_dev: tuple):
2928 )
3029 if dl_dev [0 ] != DLDeviceType .kDLOneAPI :
3130 raise ValueError ("dldevice type must be kDLOneAPI" )
32- return dpctl . SyclDevice (str (dl_dev [1 ]))
31+ return SyclDevice (str (dl_dev [1 ]))
3332
3433
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 ):
3736 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 )} "
3938 )
4039 return (DLDeviceType .kDLOneAPI , dev .get_device_id ())
You can’t perform that action at this time.
0 commit comments