From 622c472edffbe831e4e1e80c5e3168153dcdea56 Mon Sep 17 00:00:00 2001
From: Zhouheng Zheng <zhengzhouheng@outlook.com>
Date: Mon, 18 Dec 2023 22:32:21 +0800
Subject: [PATCH] Add uid() api for class operation (#668)

Type: Code Improvement
---
 include/tim/vx/operation.h | 2 ++
 src/tim/vx/operation.cc    | 1 +
 2 files changed, 3 insertions(+)

diff --git a/include/tim/vx/operation.h b/include/tim/vx/operation.h
index 0effef6..fa96377 100644
--- a/include/tim/vx/operation.h
+++ b/include/tim/vx/operation.h
@@ -48,6 +48,8 @@ class Operation {
       uint32_t accumulator_bits = 0);
   std::unique_ptr<OpImpl>& impl();
   const std::unique_ptr<OpImpl>& impl() const;
+  const uint32_t& uid() const;
+
   virtual const std::vector<std::shared_ptr<Tensor>> ConstantInputsTensor() const;
 
  protected:
diff --git a/src/tim/vx/operation.cc b/src/tim/vx/operation.cc
index 2800b21..8c4d13c 100644
--- a/src/tim/vx/operation.cc
+++ b/src/tim/vx/operation.cc
@@ -38,6 +38,7 @@ Operation::~Operation() {}
 
 std::unique_ptr<OpImpl>& Operation::impl() { return impl_; }
 const std::unique_ptr<OpImpl>& Operation::impl() const { return impl_; }
+const uint32_t& Operation::uid() const { return impl_->node()->uid; }
 
 Operation& Operation::BindInput(const std::shared_ptr<Tensor>& tensor) {
   impl_->BindInput(tensor);