From fc977c88a22217d76950e6e9071cf01bbf4f8b68 Mon Sep 17 00:00:00 2001
From: Matti Eskelinen <matti.eskelinen@amd.com>
Date: Mon, 2 Feb 2026 05:39:11 -0500
Subject: [PATCH] Add print overload for tile_distributed_index

---
 .../ck_tile/core/tensor/tile_distribution.hpp    | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

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 <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>