ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-11 17:00:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add print overload for tile_distributed_index

Browse Source
This commit is contained in:
Matti Eskelinen
2026-02-02 05:39:11 -05:00
parent 88a74ec317
commit fc977c88a2
1 changed files with 16 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
include/ck_tile/core/tensor/tile_distribution.hpp
View File

@@ -45,6 +45,22 @@ struct tile_distributed_index
CK_TILE_HOST_DEVICE static constexpr bool is_static() { return true; }
};
template <index_t... Is>
CK_TILE_HOST_DEVICE void print(const tile_distributed_index<Is...>)
{
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 <index_t... Is>