Merge elementwise op with type conversion

include/ck/tensor_operation/gpu/thread/threadwise_tensor_slice_transfer_v3r1.hpp

@@ -208,15 +208,6 @@ struct ThreadwiseTensorSliceTransfer_v3r1
             auto src_vector_container = src_vector_type{
                 src_buf.template Get<src_vector_t>(src_coord_.GetOffset(), is_src_valid)};
 
-            // apply SrcElementwiseOperation on src_vector_container
-            static_for<0, SrcScalarPerVector, 1>{}([&](auto i) {
-                SrcData src_v;
-
-                src_element_op_(src_v, src_vector_container.template AsType<SrcData>()[i]);
-
-                src_vector_container.template AsType<SrcData>()(i) = src_v;
-            });
-
             // copy data from src_vector_container into src_thread_scratch_
             src_thread_scratch_tuple_(thread_scratch_id)
                 .template SetAsType<src_vector_t>(
@@ -346,16 +337,25 @@ struct ThreadwiseTensorSliceTransfer_v3r1
                     src_vector_refs, dst_vector_refs);
             });
         }
+
         static_ford<SliceLengths>{}([&](auto idx) {
-        // pick the right conversion method
-#if CK_EXPERIMENTAL_CONVERT_PRECISION
-            using UnaryConvert = ck::tensor_operation::element_wise::UnaryConvertPrecision;
-#else
-            using UnaryConvert = ck::tensor_operation::element_wise::UnaryConvert;
-#endif
-            // convert from SrcData to DstData here
-            UnaryConvert{}(dst_thread_scratch_(idx),
-                           src_thread_scratch_tuple_[thread_scratch_id][idx]);
+            // if elementwise op does conversion, use the op instead of type_convert
+            if constexpr(is_same<SrcElementwiseOperation,
+                                 ck::tensor_operation::element_wise::UnaryConvert>::value ||
+                         is_same<SrcElementwiseOperation,
+                                 ck::tensor_operation::element_wise::UnaryConvertPrecision>::value)
+            {
+                DstData dst_v;
+                src_element_op_(dst_v, src_thread_scratch_tuple_[thread_scratch_id][idx]);
+                dst_thread_scratch_(idx) = dst_v;
+            }
+            // else apply elementwise op and use type_convert for conversion
+            else
+            {
+                SrcData src_v;
+                src_element_op_(src_v, src_thread_scratch_tuple_[thread_scratch_id][idx]);
+                dst_thread_scratch_(idx) = type_convert<DstData>(src_v);
+            }
         });
 #endif
     }