Implement csqrt

include/matx/operators/scalar_ops.h

@@ -170,8 +170,38 @@ template <typename T> struct SqrtF {
   }
 };
 
-template <typename T> using SqrtOp = UnOp<T, SqrtF<T>>;
+template <typename T>
+static __MATX_INLINE__ __MATX_HOST__ __MATX_DEVICE__ auto _internal_csqrt(T v1)
+{
+  if constexpr (is_cuda_complex_v<T>) 
+  {
+    return cuda::std::sqrt(v1);
+  }
+  else {
+    ///\todo TYLER_TODO should probably protect for only float types
+    return sqrt(static_cast<cuda::std::complex<T>>(v1));
+  }
+  if constexpr (!is_cuda_complex_v<T>) {
+    ///\todo TYLER_TODO should probably protect for only float types
+    return sqrt(static_cast<cuda::std::complex<T>>(v1));
+  }
+  else {
+    return cuda::std::sqrt(v1);
+  }  
+
+}
 
+template <typename T> struct CSqrtF {
+  static __MATX_INLINE__ std::string str() { return "csqrt"; }
+  static __MATX_INLINE__ __MATX_HOST__ __MATX_DEVICE__ auto op(T v1)
+  {
+    return _internal_csqrt(v1);
+  }
+};
+
+
+template <typename T> using SqrtOp = UnOp<T, SqrtF<T>>;
+template <typename T> using CsqrtOp = UnOp<T, CSqrtF<T>>;
 
 template <typename T>
 static __MATX_INLINE__ __MATX_HOST__ __MATX_DEVICE__ auto _internal_conj(T v1)

include/matx/operators/unary_operators.h

@@ -362,6 +362,7 @@ namespace matx
 
 #else
   DEFINE_UNARY_OP(sqrt, detail::SqrtOp);
+  DEFINE_UNARY_OP(csqrt, detail::CsqrtOp);
   DEFINE_UNARY_OP(exp, detail::ExpOp);
   DEFINE_UNARY_OP(expj, detail::ExpjOp);
   DEFINE_UNARY_OP(log10, detail::Log10Op);