We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f4859ed commit 978e551Copy full SHA for 978e551
dpctl/tensor/libtensor/include/kernels/elementwise_functions/pow.hpp
@@ -361,11 +361,20 @@ template <typename argT, typename resT> struct PowInplaceFunctor
361
tmp1 *= tmp1;
362
}
363
res = res_tmp;
364
- return;
+ }
365
+ else if constexpr (tu_ns::is_complex<argT>::value &&
366
+ tu_ns::is_complex<resT>::value)
367
+ {
368
+ using r_resT = typename resT::value_type;
369
+ using r_argT = typename argT::value_type;
370
+
371
+ res = exprm_ns::pow(exprm_ns::complex<r_resT>(res),
372
+ exprm_ns::complex<r_argT>(in));
373
374
else {
375
res = std::pow(res, in);
- };
376
377
+ return;
378
379
380
template <int vec_sz>
0 commit comments