Issue/1030: add a function to get quantization ptr

e9ac22c0 · qinyiqun · 21c6af2d · e9ac22c0
Commit e9ac22c0 authored Mar 11, 2026 by qinyiqun
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

include/infinicore/nn/linear.hpp include/infinicore/nn/linear.hpp +2 -0

No files found.
--- a/include/infinicore/nn/linear.hpp
+++ b/include/infinicore/nn/linear.hpp
@@ -35,6 +35,8 @@ public:
    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);