From e9ac22c0826e94ba9d5aa308272a78cb4841b2e8 Mon Sep 17 00:00:00 2001
From: qinyiqun <qinyiqun@outlook.com>
Date: Wed, 11 Mar 2026 16:31:50 +0800
Subject: [PATCH]  Issue/1030: add a function to get quantization ptr

---
 include/infinicore/nn/linear.hpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/include/infinicore/nn/linear.hpp b/include/infinicore/nn/linear.hpp
index 667a980fa..89a832ceb 100644
--- a/include/infinicore/nn/linear.hpp
+++ b/include/infinicore/nn/linear.hpp
@@ -35,6 +35,8 @@ class BaseLinear : public Module {
     Tensor weight_scale() const { return weight_scale_; }
     Tensor weight_zeros() const { return weight_zeros_; }
 
+    std::shared_ptr<infinicore::quantization::BaseQuantization> get_quantization() const { return quantization_; }
+
 protected:
     // Parameters
     INFINICORE_NN_PARAMETER(weight);