add complex types to oneapi::mkl::vm::abs

Author: vtavana

dpnp/backend/extensions/vm/abs.hpp

@@ -38,6 +38,18 @@ namespace ext
 {
 namespace vm
 {
+template <typename T>
+struct value_type_of
+{
+    using type = T;
+};
+
+template <typename T>
+struct value_type_of<std::complex<T>>
+{
+    using type = typename std::complex<T>::value_type;
+};
+
 template <typename T>
 sycl::event abs_contig_impl(sycl::queue exec_q,
                             const std::int64_t n,
@@ -48,7 +60,8 @@ sycl::event abs_contig_impl(sycl::queue exec_q,
     type_utils::validate_type_for_device<T>(exec_q);
 
     const T *a = reinterpret_cast<const T *>(in_a);
-    T *y = reinterpret_cast<T *>(out_y);
+    using OutT = typename value_type_of<T>::type;
+    OutT *y = reinterpret_cast<OutT *>(out_y);
 
     return mkl_vm::abs(exec_q,
                        n, // number of elements to be calculated

dpnp/backend/extensions/vm/types_matrix.hpp

@@ -53,8 +53,8 @@ template <typename T>
 struct AbsOutputType
 {
     using value_type = typename std::disjunction<
-        // TODO: Add complex type here, currently adding them here generates a
-        // compile time error due to probably a bug in mkl
+        dpctl_td_ns::TypeMapResultEntry<T, std::complex<double>>,
+        dpctl_td_ns::TypeMapResultEntry<T, std::complex<float>>,
         dpctl_td_ns::TypeMapResultEntry<T, double>,
         dpctl_td_ns::TypeMapResultEntry<T, float>,
         dpctl_td_ns::DefaultResultEntry<void>>::result_type;