Commit ff05560b authored by Rosty Geyyer's avatar Rosty Geyyer
Browse files

Format

parent 34348bd1
...@@ -347,14 +347,15 @@ struct ThreadwiseTensorSliceTransfer_v3r1 ...@@ -347,14 +347,15 @@ struct ThreadwiseTensorSliceTransfer_v3r1
}); });
} }
static_ford<SliceLengths>{}([&](auto idx) { static_ford<SliceLengths>{}([&](auto idx) {
// pick the right conversion method // pick the right conversion method
#if CK_EXPERIMENTAL_CONVERT_PRECISION #if CK_EXPERIMENTAL_CONVERT_PRECISION
using UnaryConvert = ck::tensor_operation::element_wise::UnaryConvertPrecision; using UnaryConvert = ck::tensor_operation::element_wise::UnaryConvertPrecision;
#else #else
using UnaryConvert = ck::tensor_operation::element_wise::UnaryConvert; using UnaryConvert = ck::tensor_operation::element_wise::UnaryConvert;
#endif #endif
// convert from SrcData to DstData here // convert from SrcData to DstData here
UnaryConvert{}(dst_thread_scratch_(idx), src_thread_scratch_tuple_[thread_scratch_id][idx]); UnaryConvert{}(dst_thread_scratch_(idx),
src_thread_scratch_tuple_[thread_scratch_id][idx]);
}); });
#endif #endif
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment