diff --git a/include/ck_tile/core/tensor/tile_distribution.hpp b/include/ck_tile/core/tensor/tile_distribution.hpp index f9c2aba502..aa27dcbca6 100644 --- a/include/ck_tile/core/tensor/tile_distribution.hpp +++ b/include/ck_tile/core/tensor/tile_distribution.hpp @@ -45,6 +45,22 @@ struct tile_distributed_index CK_TILE_HOST_DEVICE static constexpr bool is_static() { return true; } }; +template +CK_TILE_HOST_DEVICE void print(const tile_distributed_index) +{ + printf("tile_distributed_index<"); + if constexpr(sizeof...(Is) > 0) + { + bool first = true; + (([&first](index_t value) { + printf("%s%d", first ? "" : ", ", value); + first = false; + }(Is)), + ...); + } + printf(">"); +} + namespace detail { template