@@ -86,20 +86,23 @@ def test_get_current_device_type_outside_device_ctxt (self):
86
86
def test_get_current_device_type_inside_device_ctxt (self ):
87
87
self .assertEqual (dpctl .get_current_device_type (), None )
88
88
89
- with dpctl .device_context (dpctl . device_type . gpu ):
89
+ with dpctl .device_context ("opencl: gpu:0" ):
90
90
self .assertEqual (dpctl .get_current_device_type (), dpctl .device_type .gpu )
91
91
92
92
self .assertEqual (dpctl .get_current_device_type (), None )
93
93
94
- @unittest .skipIf (not dpctl .has_cpu_queues (), "No CPU platforms available" )
94
+ @unittest .skipUnless (dpctl .get_num_queues (backend_ty = "opencl" ,
95
+ device_ty = "cpu" ) > 0 ,
96
+ "No OpenCL CPU queues available" )
95
97
def test_get_current_device_type_inside_nested_device_ctxt (self ):
96
98
self .assertEqual (dpctl .get_current_device_type (), None )
97
99
98
- with dpctl .device_context (dpctl . device_type . cpu ):
100
+ with dpctl .device_context ("opencl: cpu:0" ):
99
101
self .assertEqual (dpctl .get_current_device_type (), dpctl .device_type .cpu )
100
102
101
- with dpctl .device_context (dpctl .device_type .gpu ):
102
- self .assertEqual (dpctl .get_current_device_type (), dpctl .device_type .gpu )
103
+ with dpctl .device_context ("opencl:gpu:0" ):
104
+ self .assertEqual (dpctl .get_current_device_type (), dpctl .
105
+ device_type .gpu )
103
106
self .assertEqual (dpctl .get_current_device_type (), dpctl .device_type .cpu )
104
107
105
108
self .assertEqual (dpctl .get_current_device_type (), None )
0 commit comments