From b1631184e315fd8dcf77a81fd293634682182d77 Mon Sep 17 00:00:00 2001
From: Yiwen Xie <yiwenx@meta.com>
Date: Fri, 6 Dec 2024 11:31:27 -0800
Subject: [PATCH] Add util function in etdump to get debug buffer size

Summary: Add util function `get_debug_buffer_size` in `ETDumpGen` to get debug buffer size in backend execution.

Differential Revision: D66855639
---
 devtools/etdump/etdump_flatcc.cpp | 4 ++++
 devtools/etdump/etdump_flatcc.h   | 1 +
 2 files changed, 5 insertions(+)

diff --git a/devtools/etdump/etdump_flatcc.cpp b/devtools/etdump/etdump_flatcc.cpp
index f34b28e650a..e175cc5d0fd 100644
--- a/devtools/etdump/etdump_flatcc.cpp
+++ b/devtools/etdump/etdump_flatcc.cpp
@@ -635,5 +635,9 @@ bool ETDumpGen::is_static_etdump() {
   return alloc_.data != nullptr;
 }
 
+size_t ETDumpGen::get_debug_buffer_size() const {
+  return debug_buffer_.size();
+}
+
 } // namespace etdump
 } // namespace executorch
diff --git a/devtools/etdump/etdump_flatcc.h b/devtools/etdump/etdump_flatcc.h
index 0bd891a0970..dcfe687577a 100644
--- a/devtools/etdump/etdump_flatcc.h
+++ b/devtools/etdump/etdump_flatcc.h
@@ -138,6 +138,7 @@ class ETDumpGen : public ::executorch::runtime::EventTracer {
       const double& output) override;
   void set_debug_buffer(::executorch::runtime::Span<uint8_t> buffer);
   ETDumpResult get_etdump_data();
+  size_t get_debug_buffer_size() const;
   size_t get_num_blocks();
   bool is_static_etdump();
   void reset();