77from options import options
88
99
10- def _sin_kernel_graph_sycl_cases (ur_adapter : str , device_architecture : str ) -> list [str ]:
10+ def _sin_kernel_graph_cases (
11+ ur_adapter : str , device_architecture : str , runtime : str
12+ ) -> list [str ]:
1113 old_adapter = options .ur_adapter
1214 old_arch = getattr (options , "device_architecture" , None )
1315 try :
@@ -16,7 +18,7 @@ def _sin_kernel_graph_sycl_cases(ur_adapter: str, device_architecture: str) -> l
1618 return [
1719 b .name ()
1820 for b in ComputeBench ().benchmarks ()
19- if "graph_api_benchmark_sycl SinKernelGraph graphs:1" in b .name ()
21+ if f"graph_api_benchmark_ { runtime } SinKernelGraph graphs:1" in b .name ()
2022 ]
2123 finally :
2224 options .ur_adapter = old_adapter
@@ -25,11 +27,11 @@ def _sin_kernel_graph_sycl_cases(ur_adapter: str, device_architecture: str) -> l
2527
2628class TestComputeBench (unittest .TestCase ):
2729 def test_sycl_sin_kernel_graph_mode_skipped_on_pvc (self ):
28- cases = _sin_kernel_graph_sycl_cases ("level_zero_v2" , "pvc" )
30+ cases = _sin_kernel_graph_cases ("level_zero_v2" , "pvc" , "sycl " )
2931 self .assertFalse (cases )
3032
3133 def test_sycl_sin_kernel_graph_mode_kept_on_non_pvc (self ):
32- cases = _sin_kernel_graph_sycl_cases ("level_zero_v2" , "bmg" )
34+ cases = _sin_kernel_graph_cases ("level_zero_v2" , "bmg" , "sycl " )
3335 self .assertEqual (
3436 sorted (cases ),
3537 sorted (
@@ -40,6 +42,10 @@ def test_sycl_sin_kernel_graph_mode_kept_on_non_pvc(self):
4042 ),
4143 )
4244
45+ def test_syclpreview_sin_kernel_graph_mode_skipped_on_pvc (self ):
46+ cases = _sin_kernel_graph_cases ("level_zero_v2" , "pvc" , "syclpreview" )
47+ self .assertFalse (cases )
48+
4349
4450if __name__ == "__main__" :
4551 unittest .main ()
0 commit comments