2222import pytest
2323
2424import dpctl
25- import dpctl .memory as dpm
2625import dpctl .compiler as dpc
26+ import dpctl .memory as dpm
2727from dpctl .compiler .utils import parse_spirv_specializations
2828
2929
@@ -42,7 +42,7 @@ def _get_level_zero_queue_or_skip():
4242
4343
4444def _skip_if_no_sycl_source_compilation (q ):
45- if not dpctl . program .is_sycl_source_compilation_available ():
45+ if not dpc .is_sycl_source_compilation_available ():
4646 pytest .skip ("SYCL source compilation extension not available" )
4747 if not q .get_sycl_device ().can_compile ("sycl" ):
4848 pytest .skip ("SYCL source compilation not supported" )
@@ -468,7 +468,7 @@ def test_create_kernel_bundle_from_sycl_source(queue_selector):
468468 }
469469 """
470470
471- prog = dpctl . program .create_kernel_bundle_from_sycl_source (
471+ prog = dpc .create_kernel_bundle_from_sycl_source (
472472 q ,
473473 sycl_source ,
474474 headers = [
@@ -479,7 +479,7 @@ def test_create_kernel_bundle_from_sycl_source(queue_selector):
479479 copts = ["-fno-fast-math" ],
480480 )
481481
482- assert type (prog ) is dpctl_prog .SyclKernelBundle
482+ assert type (prog ) is dpc .SyclKernelBundle
483483
484484 assert type (prog .addressof_ref ()) is int
485485 assert prog .has_sycl_kernel ("vector_add" )
@@ -551,10 +551,10 @@ def test_create_kernel_bundle_from_invalid_src_sycl(queue_selector):
551551 }
552552 """
553553 with pytest .raises (
554- dpctl_prog .SyclKernelBundleCompilationError ,
554+ dpc .SyclKernelBundleCompilationError ,
555555 match = "error: expected ';' at end of declaration" ,
556556 ):
557- dpctl . program .create_kernel_bundle_from_sycl_source (
557+ dpc .create_kernel_bundle_from_sycl_source (
558558 q ,
559559 sycl_source ,
560560 headers = [],
@@ -564,7 +564,7 @@ def test_create_kernel_bundle_from_invalid_src_sycl(queue_selector):
564564
565565
566566def test_sycl_source_compilation_is_available_returns_bool ():
567- v = dpctl . program .is_sycl_source_compilation_available ()
567+ v = dpc .is_sycl_source_compilation_available ()
568568 assert type (v ) is bool
569569
570570
@@ -590,9 +590,9 @@ def test_create_kernel_bundle_from_sycl_source_defaults(queue_selector):
590590 }
591591 """
592592
593- prog = dpctl . program .create_kernel_bundle_from_sycl_source (q , sycl_source )
593+ prog = dpc .create_kernel_bundle_from_sycl_source (q , sycl_source )
594594
595- assert type (prog ) is dpctl_prog .SyclKernelBundle
595+ assert type (prog ) is dpc .SyclKernelBundle
596596 assert prog .has_sycl_kernel ("vector_add" )
597597
598598
@@ -632,9 +632,7 @@ def test_create_kernel_bundle_from_sycl_source_bad_args(kwargs):
632632 # Malformed arguments must raise rather than crash, and the C API handles
633633 # allocated for the argument lists must be released on the way out.
634634 with pytest .raises (TypeError ):
635- dpctl .program .create_kernel_bundle_from_sycl_source (
636- q , sycl_source , ** kwargs
637- )
635+ dpc .create_kernel_bundle_from_sycl_source (q , sycl_source , ** kwargs )
638636
639637
640638@pytest .mark .parametrize (
@@ -666,7 +664,7 @@ def test_sycl_source_vector_add_correctness(queue_selector):
666664 }
667665 """
668666
669- prog = dpctl . program .create_kernel_bundle_from_sycl_source (
667+ prog = dpc .create_kernel_bundle_from_sycl_source (
670668 q ,
671669 sycl_source ,
672670 headers = [("math_ops.hpp" , header_content )],
0 commit comments