5145807646

# cython: language_level=3
# cython: linetrace=True
# -*- coding: utf-8 -*-
# *****************************************************************************
# Copyright (c) 2016-2023, Intel Corporation
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
# - Redistributions of source code must retain the above copyright notice,
#   this list of conditions and the following disclaimer.
# - Redistributions in binary form must reproduce the above copyright notice,
#   this list of conditions and the following disclaimer in the documentation
#   and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
# THE POSSIBILITY OF SUCH DAMAGE.
# *****************************************************************************

"""Module Backend (Searching part)

This module contains interface functions between C backend layer
and the rest of the library

"""

# NO IMPORTs here. All imports must be placed into main "dpnp_algo.pyx" file

__all__ += [
    "dpnp_argmax",
    "dpnp_argmin"
]


# C function pointer to the C library template functions
ctypedef c_dpctl.DPCTLSyclEventRef(*custom_search_1in_1out_func_ptr_t)(c_dpctl.DPCTLSyclQueueRef,
                                                                       void * , void * , size_t,
                                                                       const c_dpctl.DPCTLEventVectorRef)


cpdef utils.dpnp_descriptor dpnp_argmax(utils.dpnp_descriptor in_array1):
    cdef DPNPFuncType param1_type = dpnp_dtype_to_DPNPFuncType(in_array1.dtype)
    cdef DPNPFuncType output_type = dpnp_dtype_to_DPNPFuncType(dpnp.int64)

    cdef DPNPFuncData kernel_data = get_dpnp_function_ptr(DPNP_FN_ARGMAX_EXT, param1_type, output_type)

    in_array1_obj = in_array1.get_array()

    # ceate result array with type given by FPTR data
    cdef shape_type_c result_shape = (1,)
    cdef utils.dpnp_descriptor result = utils.create_output_descriptor(result_shape,
                                                                       kernel_data.return_type,
                                                                       None,
                                                                       device=in_array1_obj.sycl_device,
                                                                       usm_type=in_array1_obj.usm_type,
                                                                       sycl_queue=in_array1_obj.sycl_queue)

    result_sycl_queue = result.get_array().sycl_queue

    cdef c_dpctl.SyclQueue q = <c_dpctl.SyclQueue> result_sycl_queue
    cdef c_dpctl.DPCTLSyclQueueRef q_ref = q.get_queue_ref()

    cdef custom_search_1in_1out_func_ptr_t func = <custom_search_1in_1out_func_ptr_t > kernel_data.ptr

    cdef c_dpctl.DPCTLSyclEventRef event_ref = func(q_ref,
                                                    in_array1.get_data(),
                                                    result.get_data(),
                                                    in_array1.size,
                                                    NULL)  # dep_events_ref

    with nogil: c_dpctl.DPCTLEvent_WaitAndThrow(event_ref)
    c_dpctl.DPCTLEvent_Delete(event_ref)

    return result


cpdef utils.dpnp_descriptor dpnp_argmin(utils.dpnp_descriptor in_array1):
    cdef DPNPFuncType param1_type = dpnp_dtype_to_DPNPFuncType(in_array1.dtype)
    cdef DPNPFuncType output_type = dpnp_dtype_to_DPNPFuncType(dpnp.int64)

    cdef DPNPFuncData kernel_data = get_dpnp_function_ptr(DPNP_FN_ARGMIN_EXT, param1_type, output_type)

    in_array1_obj = in_array1.get_array()

    # ceate result array with type given by FPTR data
    cdef shape_type_c result_shape = (1,)
    cdef utils.dpnp_descriptor result = utils.create_output_descriptor(result_shape,
                                                                       kernel_data.return_type,
                                                                       None,
                                                                       device=in_array1_obj.sycl_device,
                                                                       usm_type=in_array1_obj.usm_type,
                                                                       sycl_queue=in_array1_obj.sycl_queue)

    result_sycl_queue = result.get_array().sycl_queue

    cdef c_dpctl.SyclQueue q = <c_dpctl.SyclQueue> result_sycl_queue
    cdef c_dpctl.DPCTLSyclQueueRef q_ref = q.get_queue_ref()

    cdef custom_search_1in_1out_func_ptr_t func = <custom_search_1in_1out_func_ptr_t > kernel_data.ptr

    cdef c_dpctl.DPCTLSyclEventRef event_ref = func(q_ref,
                                                    in_array1.get_data(),
                                                    result.get_data(),
                                                    in_array1.size,
                                                    NULL)  # dep_events_ref

    with nogil: c_dpctl.DPCTLEvent_WaitAndThrow(event_ref)
    c_dpctl.DPCTLEvent_Delete(event_ref)

    return result

1	# cython: language_level=3
2	# cython: linetrace=True
3	# -- coding: utf-8 --
4	# *****************************************************************************
5	# Copyright (c) 2016-2023, Intel Corporation
6	# All rights reserved.
7	#
8	# Redistribution and use in source and binary forms, with or without
9	# modification, are permitted provided that the following conditions are met:
10	# - Redistributions of source code must retain the above copyright notice,
11	# this list of conditions and the following disclaimer.
12	# - Redistributions in binary form must reproduce the above copyright notice,
13	# this list of conditions and the following disclaimer in the documentation
14	# and/or other materials provided with the distribution.
15	#
16	# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
17	# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18	# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19	# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
20	# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21	# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22	# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23	# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24	# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25	# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
26	# THE POSSIBILITY OF SUCH DAMAGE.
27	# *****************************************************************************
28
29	"""Module Backend (Searching part)
30
31	This module contains interface functions between C backend layer
32	and the rest of the library
33
34	"""
35
36	# NO IMPORTs here. All imports must be placed into main "dpnp_algo.pyx" file
37
38	__all__ += [	×
39	"dpnp_argmax",
40	"dpnp_argmin"
41	]
42
43
44	# C function pointer to the C library template functions
45	ctypedef c_dpctl.DPCTLSyclEventRef(*custom_search_1in_1out_func_ptr_t)(c_dpctl.DPCTLSyclQueueRef,
46	void * , void * , size_t,
47	const c_dpctl.DPCTLEventVectorRef)
48
49
UNCOV 50	cpdef utils.dpnp_descriptor dpnp_argmax(utils.dpnp_descriptor in_array1):	×
51	cdef DPNPFuncType param1_type = dpnp_dtype_to_DPNPFuncType(in_array1.dtype)	1✔
52	cdef DPNPFuncType output_type = dpnp_dtype_to_DPNPFuncType(dpnp.int64)	1✔
53
54	cdef DPNPFuncData kernel_data = get_dpnp_function_ptr(DPNP_FN_ARGMAX_EXT, param1_type, output_type)	1✔
55
56	in_array1_obj = in_array1.get_array()	1✔
57
58	# ceate result array with type given by FPTR data
59	cdef shape_type_c result_shape = (1,)	1✔
60	cdef utils.dpnp_descriptor result = utils.create_output_descriptor(result_shape,	1✔
61	kernel_data.return_type,
62	None,
63	device=in_array1_obj.sycl_device,	1✔
64	usm_type=in_array1_obj.usm_type,	1✔
65	sycl_queue=in_array1_obj.sycl_queue)	1✔
66
67	result_sycl_queue = result.get_array().sycl_queue	1✔
68
69	cdef c_dpctl.SyclQueue q = <c_dpctl.SyclQueue> result_sycl_queue	1✔
70	cdef c_dpctl.DPCTLSyclQueueRef q_ref = q.get_queue_ref()	1✔
71
72	cdef custom_search_1in_1out_func_ptr_t func = <custom_search_1in_1out_func_ptr_t > kernel_data.ptr	1✔
73
74	cdef c_dpctl.DPCTLSyclEventRef event_ref = func(q_ref,	1✔
75	in_array1.get_data(),
76	result.get_data(),
77	in_array1.size,	1✔
78	NULL) # dep_events_ref
79
80	with nogil: c_dpctl.DPCTLEvent_WaitAndThrow(event_ref)	1✔
81	c_dpctl.DPCTLEvent_Delete(event_ref)	1✔
82
83	return result	1✔
84
85
86	cpdef utils.dpnp_descriptor dpnp_argmin(utils.dpnp_descriptor in_array1):	×
87	cdef DPNPFuncType param1_type = dpnp_dtype_to_DPNPFuncType(in_array1.dtype)	1✔
88	cdef DPNPFuncType output_type = dpnp_dtype_to_DPNPFuncType(dpnp.int64)	1✔
89
90	cdef DPNPFuncData kernel_data = get_dpnp_function_ptr(DPNP_FN_ARGMIN_EXT, param1_type, output_type)	1✔
91
92	in_array1_obj = in_array1.get_array()	1✔
93
94	# ceate result array with type given by FPTR data
95	cdef shape_type_c result_shape = (1,)	1✔
96	cdef utils.dpnp_descriptor result = utils.create_output_descriptor(result_shape,	1✔
97	kernel_data.return_type,
98	None,
99	device=in_array1_obj.sycl_device,	1✔
100	usm_type=in_array1_obj.usm_type,	1✔
101	sycl_queue=in_array1_obj.sycl_queue)	1✔
102
103	result_sycl_queue = result.get_array().sycl_queue	1✔
104
105	cdef c_dpctl.SyclQueue q = <c_dpctl.SyclQueue> result_sycl_queue	1✔
106	cdef c_dpctl.DPCTLSyclQueueRef q_ref = q.get_queue_ref()	1✔
107
108	cdef custom_search_1in_1out_func_ptr_t func = <custom_search_1in_1out_func_ptr_t > kernel_data.ptr	1✔
109
110	cdef c_dpctl.DPCTLSyclEventRef event_ref = func(q_ref,	1✔
111	in_array1.get_data(),
112	result.get_data(),
113	in_array1.size,	1✔
114	NULL) # dep_events_ref
115
116	with nogil: c_dpctl.DPCTLEvent_WaitAndThrow(event_ref)	1✔
117	c_dpctl.DPCTLEvent_Delete(event_ref)	1✔
118
119	return result	1✔

IntelPython / dpnp / 5145807646

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