print.hpp Source File

print.hpp Source File#

Composable Kernel: print.hpp Source File
// SPDX-License-Identifier: MIT
// Copyright (c) 2025, Advanced Micro Devices, Inc. All rights reserved.
 
#pragma once
 
#include "ck_tile/core/config.hpp"
 
namespace ck_tile {
template <typename T>
CK_TILE_HOST_DEVICE void print(const T&)
{
    static_assert(sizeof(T) == 0,
                  "No print implementation available for this type. Please specialize "
                  "ck_tile::print for your type.");
}
template <>
CK_TILE_HOST_DEVICE void print(const int& value)
{
    printf("%d", value);
}
template <>
CK_TILE_HOST_DEVICE void print(const float& value)
{
    printf("%f", value);
}
template <>
CK_TILE_HOST_DEVICE void print(const double& value)
{
    printf("%f", value);
}
template <>
CK_TILE_HOST_DEVICE void print(const long& value)
{
    printf("%ld", value);
}
template <>
CK_TILE_HOST_DEVICE void print(const unsigned int& value)
{
    printf("%u", value);
}
template <>
CK_TILE_HOST_DEVICE void print(const char& value)
{
    printf("%c", value);
}
template <typename T, size_t N>
CK_TILE_HOST_DEVICE void print(const T (&value)[N])
{
    printf("[");
    for(size_t i = 0; i < N; ++i)
    {
        if(i > 0)
            printf(", ");
        print(value[i]); // Recursively call print for each element
    }
    printf("]");
}
 
} // namespace ck_tile