@@ -124,6 +124,34 @@ PYBIND11_MODULE(_vm_impl, m)
124
124
py::arg (" sycl_queue" ), py::arg (" src" ), py::arg (" dst" ));
125
125
}
126
126
127
+ // UnaryUfunc: ==== Conj(x) ====
128
+ {
129
+ vm_ext::init_ufunc_dispatch_vector<unary_impl_fn_ptr_t ,
130
+ vm_ext::ConjContigFactory>(
131
+ conj_dispatch_vector);
132
+
133
+ auto conj_pyapi = [&](sycl::queue exec_q, arrayT src, arrayT dst,
134
+ const event_vecT &depends = {}) {
135
+ return vm_ext::unary_ufunc (exec_q, src, dst, depends,
136
+ conj_dispatch_vector);
137
+ };
138
+ m.def (" _conj" , conj_pyapi,
139
+ " Call `conj` function from OneMKL VM library to compute "
140
+ " conjugate of vector elements" ,
141
+ py::arg (" sycl_queue" ), py::arg (" src" ), py::arg (" dst" ),
142
+ py::arg (" depends" ) = py::list ());
143
+
144
+ auto conj_need_to_call_pyapi = [&](sycl::queue exec_q, arrayT src,
145
+ arrayT dst) {
146
+ return vm_ext::need_to_call_unary_ufunc (exec_q, src, dst,
147
+ conj_dispatch_vector);
148
+ };
149
+ m.def (" _mkl_conj_to_call" , conj_need_to_call_pyapi,
150
+ " Check input arguments to answer if `conj` function from "
151
+ " OneMKL VM library can be used" ,
152
+ py::arg (" sycl_queue" ), py::arg (" src" ), py::arg (" dst" ));
153
+ }
154
+
127
155
// UnaryUfunc: ==== Cos(x) ====
128
156
{
129
157
vm_ext::init_ufunc_dispatch_vector<unary_impl_fn_ptr_t ,
@@ -180,34 +208,6 @@ PYBIND11_MODULE(_vm_impl, m)
180
208
py::arg (" sycl_queue" ), py::arg (" src" ), py::arg (" dst" ));
181
209
}
182
210
183
- // UnaryUfunc: ==== Conj(x) ====
184
- {
185
- vm_ext::init_ufunc_dispatch_vector<unary_impl_fn_ptr_t ,
186
- vm_ext::ConjContigFactory>(
187
- conj_dispatch_vector);
188
-
189
- auto conj_pyapi = [&](sycl::queue exec_q, arrayT src, arrayT dst,
190
- const event_vecT &depends = {}) {
191
- return vm_ext::unary_ufunc (exec_q, src, dst, depends,
192
- conj_dispatch_vector);
193
- };
194
- m.def (" _conj" , conj_pyapi,
195
- " Call `conj` function from OneMKL VM library to compute "
196
- " conjugate of vector elements" ,
197
- py::arg (" sycl_queue" ), py::arg (" src" ), py::arg (" dst" ),
198
- py::arg (" depends" ) = py::list ());
199
-
200
- auto conj_need_to_call_pyapi = [&](sycl::queue exec_q, arrayT src,
201
- arrayT dst) {
202
- return vm_ext::need_to_call_unary_ufunc (exec_q, src, dst,
203
- conj_dispatch_vector);
204
- };
205
- m.def (" _mkl_conj_to_call" , conj_need_to_call_pyapi,
206
- " Check input arguments to answer if `conj` function from "
207
- " OneMKL VM library can be used" ,
208
- py::arg (" sycl_queue" ), py::arg (" src" ), py::arg (" dst" ));
209
- }
210
-
211
211
// UnaryUfunc: ==== Ln(x) ====
212
212
{
213
213
vm_ext::init_ufunc_dispatch_vector<unary_impl_fn_ptr_t ,
0 commit comments