Fix forge types missing formatter

Adding formatter for forge types to enable debug build without any hacks.
tenstorrent · Aug 31, 2024 · 8ec4a13 · 8ec4a13
1 parent 50251d3
commit 8ec4a13
Show file tree

Hide file tree

Showing 4 changed files with 50 additions and 0 deletions.
diff --git a/pybuda/csrc/backend_api/device_config.hpp b/pybuda/csrc/backend_api/device_config.hpp
@@ -452,3 +452,13 @@ inline std::ostream& operator<<(std::ostream& os, DeviceConfig const& device_con
 }
 
 }  // namespace tt
+
+template<> struct fmt::formatter<tt::DeviceConfig> : fmt::formatter<std::string_view>
+{
+    inline auto format(const tt::DeviceConfig& device_config, fmt::format_context& ctx) const -> decltype(ctx.out())
+    {
+        std::ostringstream oss;
+        oss << device_config;
+        return fmt::formatter<std::string_view>::format(oss.str(), ctx);
+    }
+};
diff --git a/pybuda/csrc/graph_lib/node_types.hpp b/pybuda/csrc/graph_lib/node_types.hpp
@@ -655,3 +655,13 @@ inline std::string to_string(InputNodeType t)
 
 }  // namespace graphlib
 }  // namespace tt
+
+template<> struct fmt::formatter<tt::graphlib::OpType> : fmt::formatter<std::string_view>
+{
+    inline auto format(const tt::graphlib::OpType& op_type, fmt::format_context& ctx) const -> decltype(ctx.out())
+    {
+        std::ostringstream oss;
+        oss << op_type;
+        return fmt::formatter<std::string_view>::format(oss.str(), ctx);
+    }
+};
diff --git a/pybuda/csrc/passes/nd_slice.hpp b/pybuda/csrc/passes/nd_slice.hpp
@@ -296,3 +296,13 @@ inline std::ostream& operator<<(std::ostream& out, NDSlice::Slice const& c)
     return out;
 }
 }  // namespace tt::passes
+
+template<> struct fmt::formatter<tt::passes::NDSlice> : fmt::formatter<std::string_view>
+{
+    inline auto format(const tt::passes::NDSlice& nd_slice, fmt::format_context& ctx) const -> decltype(ctx.out())
+    {
+        std::ostringstream oss;
+        oss << nd_slice;
+        return fmt::formatter<std::string_view>::format(oss.str(), ctx);
+    }
+};
diff --git a/pybuda/csrc/tt_torch_device/torch_device_impl.cpp b/pybuda/csrc/tt_torch_device/torch_device_impl.cpp
@@ -554,3 +554,23 @@ TORCH_LIBRARY_IMPL(_, PrivateUse1, m)
     m.fallback(torch::CppFunction::makeFromBoxedFunction<&tt::fallback>());
 }
 
+template<> struct fmt::formatter<c10::Device> : fmt::formatter<std::string_view>
+{
+    inline auto format(const c10::Device& device, fmt::format_context& ctx) const -> decltype(ctx.out())
+    {
+        std::ostringstream oss;
+        oss << device;
+        return fmt::formatter<std::string_view>::format(oss.str(), ctx);
+    }
+};
+
+template<> struct fmt::formatter<c10::OperatorName> : fmt::formatter<std::string_view>
+{
+    inline auto format(const c10::OperatorName& operator_name, fmt::format_context& ctx) const -> decltype(ctx.out())
+    {
+        std::ostringstream oss;
+        oss << operator_name;
+        return fmt::formatter<std::string_view>::format(oss.str(), ctx);
+    }
+};
+