MCPcopy Create free account
hub / github.com/NVIDIA/cutlass / transform

Function transform

include/cute/algorithm/tuple_algorithms.hpp:205–216  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

203
204template <class T, class F>
205CUTE_HOST_DEVICE constexpr
206auto
207transform(T const& t, F&& f)
208{
209 if constexpr (is_tuple<T>::value) {
210 return detail::tapply(t, f, [](auto const&... a){ return cute::make_tuple(a...); }, tuple_seq<T>{});
211 } else {
212 return f(t);
213 }
214
215 CUTE_GCC_UNREACHABLE;
216}
217
218template <class T0, class T1, class F>
219CUTE_HOST_DEVICE constexpr

Callers 15

epilogue_no_predicationFunction · 0.70
copy_ifFunction · 0.70
transform_leafFunction · 0.70
repeat_likeFunction · 0.70
dequantize_kernelFunction · 0.50
TensorTransformReduceFunction · 0.50
TensorTransformReduceFunction · 0.50
TESTFunction · 0.50
make_stride_tupleMethod · 0.50
storeMethod · 0.50

Calls 2

tapplyFunction · 0.70
fFunction · 0.50

Tested by 15

storeMethod · 0.40
storeMethod · 0.40
transformMethod · 0.40
accumMethod · 0.40
transformMethod · 0.40
accumMethod · 0.40
transformMethod · 0.40
accumMethod · 0.40
transformMethod · 0.40
accumMethod · 0.40
end_loopMethod · 0.40
endMethod · 0.40