7267921802

Committed 19 Dec 2023 09:45PM UTC coverage: 78.889% (-0.8%) from 79.663%

Build # 7267921802

Build Type

Pull #1253

github

Committed by

web-flow

Commit Message

Merge c7a5862f7 into 6b76fd3de

Pull Request Pull Request #1253: WIP: Experimental/barrier functions

Run Details

1222 of 1820 branches covered (0.0%)

Branch coverage included in aggregate %.

50 of 112 new or added lines in 6 files covered. (44.64%)

1 existing line in 1 file now uncovered.

5437 of 6621 relevant lines covered (82.12%)

0.82 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

34.04

/numba_dpex/experimental/kernel_iface/dpcpp_spirv_overloads/_barriers.py

# SPDX-FileCopyrightText: 2023 Intel Corporation
#
# SPDX-License-Identifier: Apache-2.0

"""
Provides overloads for functions included in kernel_iface.barrier that
generate dpcpp SPIR-V LLVM IR intrinsic function calls.
"""
from llvmlite import ir as llvmir
from numba.core import cgutils, types
from numba.extending import intrinsic, overload

from numba_dpex.core import itanium_mangler as ext_itanium_mangler
from numba_dpex.experimental.kernel_iface import (
    group_barrier,
    sub_group_barrier,
)
from numba_dpex.experimental.target import DPEX_KERNEL_EXP_TARGET_NAME

from ..memory_enums import MemoryOrder, MemoryScope
from ._spv_inst_helper import get_memory_semantics_mask, get_scope


def get_memory_scope(fence_scope):
    if isinstance(fence_scope, types.Literal):
        return get_scope(fence_scope.literal_value)
    else:
        return get_scope(fence_scope.value)


@intrinsic
def _intrinsic_barrier(
    ty_context, ty_exec_scope, ty_mem_scope, ty_spirv_mem_sem_mask
):
    sig = types.void(ty_exec_scope, ty_mem_scope, ty_spirv_mem_sem_mask)

    def _intrinsic_barrier_codegen(context, builder, sig, args):
        fn_name = "__spirv_ControlBarrier"
        mangled_fn_name = ext_itanium_mangler.mangle_ext(
            fn_name, [ty_exec_scope, ty_mem_scope, ty_spirv_mem_sem_mask]
        )

        spirv_fn_arg_types = [context.get_value_type(t) for t in sig.args]

        fnty = llvmir.FunctionType(llvmir.VoidType(), spirv_fn_arg_types)

        fn_args = [args[0], args[1], args[2]]

        fn = cgutils.get_or_insert_function(
            builder.module, fnty, mangled_fn_name
        )
        # XXX Uncomment once llvmlite PR#1019 is merged and available for use
        # fn.attributes.add("convergent")
        # fn.attributes.add("nounwind")
        fn.calling_convention = "spir_func"

        builder.call(fn, fn_args)

        # XXX Uncomment once llvmlite PR#1019 is merged and available for use
        # callinst.attributes.add("convergent")
        # callinst.attributes.add("nounwind")

        return

    return (
        sig,
        _intrinsic_barrier_codegen,
    )


@overload(
    group_barrier,
    prefer_literal=True,
    target=DPEX_KERNEL_EXP_TARGET_NAME,
)
def _ol_group_barrier(fence_scope=MemoryScope.WORK_GROUP):
    spirv_memory_semantics_mask = get_memory_semantics_mask(
        MemoryOrder.SEQ_CST.value
    )
    exec_scope = get_scope(MemoryScope.WORK_GROUP.value)
    mem_scope = get_memory_scope(fence_scope)

    def _ol_group_barrier_impl(fence_scope=MemoryScope.WORK_GROUP):
        return _intrinsic_barrier(
            exec_scope, mem_scope, spirv_memory_semantics_mask
        )

    return _ol_group_barrier_impl


@overload(
    sub_group_barrier,
    prefer_literal=True,
    target=DPEX_KERNEL_EXP_TARGET_NAME,
)
def _ol_sub_group_barrier(fence_scope=MemoryScope.SUB_GROUP):
    spirv_memory_semantics_mask = get_memory_semantics_mask(
        MemoryOrder.SEQ_CST.value
    )
    exec_scope = get_scope(MemoryScope.SUB_GROUP.value)
    mem_scope = get_memory_scope(fence_scope)

    def _ol_sub_group_barrier_impl(fence_scope=MemoryScope.SUB_GROUP):
        return _intrinsic_barrier(
            exec_scope, mem_scope, spirv_memory_semantics_mask
        )

    return _ol_sub_group_barrier_impl

1	# SPDX-FileCopyrightText: 2023 Intel Corporation
2	#
3	# SPDX-License-Identifier: Apache-2.0
4
5	"""	1✔
6	Provides overloads for functions included in kernel_iface.barrier that
7	generate dpcpp SPIR-V LLVM IR intrinsic function calls.
8	"""
9	from llvmlite import ir as llvmir	1✔
10	from numba.core import cgutils, types	1✔
11	from numba.extending import intrinsic, overload	1✔
12
13	from numba_dpex.core import itanium_mangler as ext_itanium_mangler	1✔
14	from numba_dpex.experimental.kernel_iface import (	1✔
15	group_barrier,
16	sub_group_barrier,
17	)
18	from numba_dpex.experimental.target import DPEX_KERNEL_EXP_TARGET_NAME	1✔
19
20	from ..memory_enums import MemoryOrder, MemoryScope	1✔
21	from ._spv_inst_helper import get_memory_semantics_mask, get_scope	1✔
22
23
24	def get_memory_scope(fence_scope):	1✔
NEW 25	if isinstance(fence_scope, types.Literal):	×
NEW 26	return get_scope(fence_scope.literal_value)	×
27	else:
NEW 28	return get_scope(fence_scope.value)	×
29
30
31	@intrinsic	1✔
32	def _intrinsic_barrier(	1✔
33	ty_context, ty_exec_scope, ty_mem_scope, ty_spirv_mem_sem_mask
34	):
NEW 35	sig = types.void(ty_exec_scope, ty_mem_scope, ty_spirv_mem_sem_mask)	×
36
NEW 37	def _intrinsic_barrier_codegen(context, builder, sig, args):	×
NEW 38	fn_name = "__spirv_ControlBarrier"	×
NEW 39	mangled_fn_name = ext_itanium_mangler.mangle_ext(	×
40	fn_name, [ty_exec_scope, ty_mem_scope, ty_spirv_mem_sem_mask]
41	)
42
NEW 43	spirv_fn_arg_types = [context.get_value_type(t) for t in sig.args]	×
44
NEW 45	fnty = llvmir.FunctionType(llvmir.VoidType(), spirv_fn_arg_types)	×
46
NEW 47	fn_args = [args[0], args[1], args[2]]	×
48
NEW 49	fn = cgutils.get_or_insert_function(	×
50	builder.module, fnty, mangled_fn_name
51	)
52	# XXX Uncomment once llvmlite PR#1019 is merged and available for use
53	# fn.attributes.add("convergent")
54	# fn.attributes.add("nounwind")
NEW 55	fn.calling_convention = "spir_func"	×
56
NEW 57	builder.call(fn, fn_args)	×
58
59	# XXX Uncomment once llvmlite PR#1019 is merged and available for use
60	# callinst.attributes.add("convergent")
61	# callinst.attributes.add("nounwind")
62
NEW 63	return	×
64
NEW 65	return (	×
66	sig,
67	_intrinsic_barrier_codegen,
68	)
69
70
71	@overload(	1✔
72	group_barrier,
73	prefer_literal=True,
74	target=DPEX_KERNEL_EXP_TARGET_NAME,
75	)
76	def _ol_group_barrier(fence_scope=MemoryScope.WORK_GROUP):	1✔
NEW 77	spirv_memory_semantics_mask = get_memory_semantics_mask(	×
78	MemoryOrder.SEQ_CST.value
79	)
NEW 80	exec_scope = get_scope(MemoryScope.WORK_GROUP.value)	×
NEW 81	mem_scope = get_memory_scope(fence_scope)	×
82
NEW 83	def _ol_group_barrier_impl(fence_scope=MemoryScope.WORK_GROUP):	×
NEW 84	return _intrinsic_barrier(	×
85	exec_scope, mem_scope, spirv_memory_semantics_mask
86	)
87
NEW 88	return _ol_group_barrier_impl	×
89
90
91	@overload(	1✔
92	sub_group_barrier,
93	prefer_literal=True,
94	target=DPEX_KERNEL_EXP_TARGET_NAME,
95	)
96	def _ol_sub_group_barrier(fence_scope=MemoryScope.SUB_GROUP):	1✔
NEW 97	spirv_memory_semantics_mask = get_memory_semantics_mask(	×
98	MemoryOrder.SEQ_CST.value
99	)
NEW 100	exec_scope = get_scope(MemoryScope.SUB_GROUP.value)	×
NEW 101	mem_scope = get_memory_scope(fence_scope)	×
102
NEW 103	def _ol_sub_group_barrier_impl(fence_scope=MemoryScope.SUB_GROUP):	×
NEW 104	return _intrinsic_barrier(	×
105	exec_scope, mem_scope, spirv_memory_semantics_mask
106	)
107
NEW 108	return _ol_sub_group_barrier_impl	×

IntelPython / numba-dpex / 7267921802

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous