27
27
import unittest
28
28
29
29
class TestGetNumPlatforms (unittest .TestCase ):
30
- @unittest .skipIf (not dppl .has_sycl_platforms , "No SYCL platforms available" )
30
+ @unittest .skipIf (not dppl .has_sycl_platforms () , "No SYCL platforms available" )
31
31
def test_dppl_get_num_platforms (self ):
32
32
if (dppl .has_sycl_platforms ):
33
33
self .assertGreaterEqual (dppl .get_num_platforms (), 1 )
34
34
35
- @unittest .skipIf (not dppl .has_sycl_platforms , "No SYCL platforms available" )
35
+ @unittest .skipIf (not dppl .has_sycl_platforms () , "No SYCL platforms available" )
36
36
class TestDumpMethods (unittest .TestCase ):
37
37
def test_dppl_dump (self ):
38
38
try :
@@ -47,7 +47,7 @@ def test_dppl_dump_device_info (self):
47
47
except Exception :
48
48
self .fail ("Encountered an exception inside dump_device_info()." )
49
49
50
- @unittest .skipIf (not dppl .has_sycl_platforms , "No SYCL platforms available" )
50
+ @unittest .skipIf (not dppl .has_sycl_platforms () , "No SYCL platforms available" )
51
51
class TestDPPLIsInDPPLCtxt (unittest .TestCase ):
52
52
53
53
def test_is_in_dppl_ctxt_outside_device_ctxt (self ):
@@ -57,28 +57,31 @@ def test_is_in_dppl_ctxt_inside_device_ctxt (self):
57
57
with dppl .device_context (dppl .device_type .gpu ):
58
58
self .assertTrue (dppl .is_in_dppl_ctxt ())
59
59
60
+ @unittest .skipIf (not dppl .has_cpu_queues (), "No CPU platforms available" )
60
61
def test_is_in_dppl_ctxt_inside_nested_device_ctxt (self ):
61
62
with dppl .device_context (dppl .device_type .cpu ):
62
63
with dppl .device_context (dppl .device_type .gpu ):
63
64
self .assertTrue (dppl .is_in_dppl_ctxt ())
64
65
self .assertTrue (dppl .is_in_dppl_ctxt ())
65
66
self .assertFalse (dppl .is_in_dppl_ctxt ())
66
67
67
- @unittest .skipIf (not dppl .has_sycl_platforms , "No SYCL platforms available" )
68
+ @unittest .skipIf (not dppl .has_sycl_platforms () , "No SYCL platforms available" )
68
69
class TestGetCurrentQueueInMultipleThreads (unittest .TestCase ):
69
70
70
71
def test_num_current_queues_outside_with_clause (self ):
71
72
self .assertEqual (dppl .get_num_activated_queues (), 0 )
72
73
73
- @unittest .skipIf (not dppl .has_gpu_queues , "No GPU platforms available" )
74
+ @unittest .skipIf (not dppl .has_gpu_queues (), "No GPU platforms available" )
75
+ @unittest .skipIf (not dppl .has_cpu_queues (), "No CPU platforms available" )
74
76
def test_num_current_queues_inside_with_clause (self ):
75
77
with dppl .device_context (dppl .device_type .cpu ):
76
78
self .assertEqual (dppl .get_num_activated_queues (), 1 )
77
79
with dppl .device_context (dppl .device_type .gpu ):
78
80
self .assertEqual (dppl .get_num_activated_queues (), 2 )
79
81
self .assertEqual (dppl .get_num_activated_queues (), 0 )
80
82
81
- @unittest .skipIf (not dppl .has_gpu_queues , "No GPU platforms available" )
83
+ @unittest .skipIf (not dppl .has_gpu_queues (), "No GPU platforms available" )
84
+ @unittest .skipIf (not dppl .has_cpu_queues (), "No CPU platforms available" )
82
85
def test_num_current_queues_inside_threads (self ):
83
86
from threading import Thread , local
84
87
def SessionThread (self ):
0 commit comments