Skip to content

Commit b181cd1

Browse files
cccclaifacebook-github-bot
authored andcommitted
Add et_dump related API to module (#9284)
Summary: Pull Request resolved: #9284 It is for reusing the exention/module in the module definition in pybindings https://github.com/pytorch/executorch/blob/1a918c779e16c0ee903a08b30c1c666d1efb2c57/extension/pybindings/pybindings.cpp#L172 Differential Revision: D71135352
1 parent 30582ab commit b181cd1

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

extension/module/module.h

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -478,6 +478,16 @@ class Module {
478478
return event_tracer_.get();
479479
}
480480

481+
bool has_etdump_debug_buffer() const {
482+
return debug_buffer_.empty() == false;
483+
}
484+
485+
ET_NODISCARD
486+
runtime::Span<uint8_t> get_etdump_debug_buffer() {
487+
return runtime::Span<uint8_t>(debug_buffer_.data(), debug_buffer_.size());
488+
}
489+
490+
481491
private:
482492
struct MethodHolder {
483493
std::vector<std::vector<uint8_t>> planned_buffers;
@@ -498,6 +508,8 @@ class Module {
498508
std::unique_ptr<runtime::EventTracer> event_tracer_;
499509
std::unique_ptr<runtime::DataLoader> data_map_loader_;
500510
std::unique_ptr<NamedDataMap> data_map_;
511+
std::vector<uint8_t> debug_buffer_;
512+
size_t debug_buffer_size_;
501513

502514
protected:
503515
std::unordered_map<std::string, MethodHolder> methods_;

0 commit comments

Comments
 (0)