File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed
dpctl/tensor/libtensor/include/kernels/elementwise_functions Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -53,14 +53,16 @@ using dpctl::tensor::ssize_t;
5353namespace td_ns = dpctl::tensor::type_dispatch;
5454
5555using dpctl::tensor::type_utils::is_complex;
56+ using dpctl::tensor::type_utils::is_complex_v;
5657
5758template <typename argT, typename resT> struct ImagFunctor
5859{
5960
6061 // is function constant for given argT
61- using is_constant = typename std::false_type;
62+ using is_constant =
63+ typename std::is_same<is_complex<argT>, std::false_type>;
6264 // constant value, if constant
63- // constexpr resT constant_value = resT{};
65+ static constexpr resT constant_value = resT{0 };
6466 // is function defined for sycl::vec
6567 using supports_vec = typename std::false_type;
6668 // do both argTy and resTy support sugroup store/load operation
@@ -74,7 +76,7 @@ template <typename argT, typename resT> struct ImagFunctor
7476 }
7577 else {
7678 static_assert (std::is_same_v<resT, argT>);
77- return resT{ 0 } ;
79+ return constant_value ;
7880 }
7981 }
8082};
You can’t perform that action at this time.
0 commit comments