concat.hpp Source File

concat.hpp Source File#

Composable Kernel: concat.hpp Source File
concat.hpp
Go to the documentation of this file.
1// SPDX-License-Identifier: MIT
2// Copyright (c) 2025, Advanced Micro Devices, Inc. All rights reserved.
3
4#pragma once
5
6#include "ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_scheduler.hpp"
7
8namespace ck_tile {
9
10template <typename T>
11struct IsCharArray : std::false_type
12{
13};
14
15template <std::size_t N>
16struct IsCharArray<char[N]> : std::true_type
17{
18};
19
20template <std::size_t N>
21struct IsCharArray<const char[N]> : std::true_type
22{
23};
24
25template <std::size_t N>
26struct IsCharArray<char (&)[N]> : std::true_type
27{
28};
29
30template <std::size_t N>
31struct IsCharArray<const char (&)[N]> : std::true_type
32{
33};
34
35template <typename... Ts>
36inline constexpr bool AllConvertibleToStringView =
37 ((std::is_convertible_v<Ts, std::string_view> || IsCharArray<Ts>::value ||
38 std::is_same_v<Ts, char>) &&
39 ...);
40
41template <typename... Ts>
42[[nodiscard]] auto
43concat(const Ts&... xs) -> std::enable_if_t<!AllConvertibleToStringView<Ts...>, std::string>
44{
45 using ::operator<<;
46 thread_local std::ostringstream oss;
47 oss.str("");
48
49 (oss << ... << xs);
50 return oss.str();
51}
52
53template <std::size_t N>
54[[nodiscard]] constexpr inline std::size_t getSize(char (&)[N]) noexcept
55{
56 return N;
57}
58
59template <std::size_t N>
60[[nodiscard]] constexpr inline std::size_t getSize(const char (&)[N]) noexcept
61{
62 return N;
63}
64
65[[nodiscard]] constexpr inline std::size_t getSize(const char* s) noexcept
66{
67 const char* end = s;
68 while(*end++ != 0) {}
69 return end - s - 1;
70}
71
72[[nodiscard]] constexpr inline std::size_t getSize(const char&) noexcept { return 1; }
73
74[[nodiscard]] inline std::size_t getSize(const std::string& s) noexcept { return s.size(); }
75
76[[nodiscard]] constexpr inline std::size_t getSize(const std::string_view& s) noexcept
77{
78 return s.size();
79}
80
81template <typename... Ts>
82auto concatInto(std::string& result,
83 const Ts&... xs) -> std::enable_if_t<AllConvertibleToStringView<Ts...>, void>
84{
85 const std::size_t space = (1 + ... + getSize(xs));
86 result.reserve(result.size() + space);
87 ((result += xs), ...);
88}
89
90template <typename... Ts>
91[[nodiscard]] auto
92concat(const Ts&... xs) -> std::enable_if_t<AllConvertibleToStringView<Ts...>, std::string>
93{
94 std::string result;
95 concatInto(result, xs...);
96 return result;
97}
98
99// Function for types convertible to std::string_view
100template <typename Sep, typename First, typename... Rest>
101[[nodiscard]] auto concat(Sep sep, const First& first, const Rest&... rest)
102 -> std::enable_if_t<AllConvertibleToStringView<First, Rest...>, std::string>
103{
104 std::string result;
105 result += first;
106 ((result += sep, result += rest), ...);
107 return result;
108}
109
110// Function for other types
111template <typename Sep, typename First, typename... Rest>
112[[nodiscard]] auto concat(Sep sep, const First& first, const Rest&... rest)
113 -> std::enable_if_t<!AllConvertibleToStringView<First, Rest...>, std::string>
114{
115 using ::operator<<;
116 thread_local std::ostringstream oss;
117 oss.str("");
118 oss << first;
119 ((oss << sep << rest), ...);
120 return oss.str();
121}
122
123} // namespace ck_tile
gemm_pipeline_ag_bg_cr_scheduler.hpp
ck_tile
Definition tile/core/algorithm/cluster_descriptor.hpp:13
ck_tile::concatInto
auto concatInto(std::string &result, const Ts &... xs) -> std::enable_if_t< AllConvertibleToStringView< Ts... >, void >
Definition concat.hpp:82
ck_tile::AllConvertibleToStringView
constexpr bool AllConvertibleToStringView
Definition concat.hpp:36
ck_tile::concat
auto concat(const Ts &... xs) -> std::enable_if_t<!AllConvertibleToStringView< Ts... >, std::string >
Definition concat.hpp:43
ck_tile::getSize
constexpr std::size_t getSize(char(&)[N]) noexcept
Definition concat.hpp:54
std
STL namespace.
ck_tile::IsCharArray
Definition concat.hpp:12