feat(mgb/opr): add property params for python operator class

GitOrigin-RevId: af6da0e0ac

python_module/src/cpp/megbrain_wrap.cpp

@@ -880,6 +880,11 @@ SymbolVar SharedScalar::_as_sym_var(CompGraph &cg, mgb::CompNode &cn) {
             ssprintf("SharedScalar@%p", m_val.get()));
 }
 
+/* =============== Operator ===============  */
+
+const std::unique_ptr<mgb::OprFootprint> Operator::sm_opr_footprint_ptr{
+                                std::make_unique<mgb::OprFootprint>()};
+
 /* ================= misc =================  */
 
 SymbolVar fill_retain_dtype(SymbolVar var, PyObject *value) {

python_module/src/cpp/megbrain_wrap.h

@@ -17,6 +17,8 @@
 #include "megbrain/graph.h"
 #include "megbrain/opr/io.h"
 
+#include "megbrain/plugin/opr_footprint.h"
+
 #include <map>
 #include <string>
 
@@ -441,16 +443,24 @@ class SharedScalar {
  */
 class Operator {
     mgb::cg::OperatorNodeBase* m_operator_node;
+    std::string m_params;
+
+    static const std::unique_ptr<mgb::OprFootprint> sm_opr_footprint_ptr;
 
 public:
     Operator() : m_operator_node(nullptr){};
     Operator(mgb::cg::OperatorNodeBase* operator_node)
-            : m_operator_node(operator_node) {}
+            : m_operator_node(operator_node),
+            m_params(std::move(
+                        (sm_opr_footprint_ptr->calc_footprint(m_operator_node)).param->to_string()))
+            {}
 
     size_t id() const { return m_operator_node->id(); }
 
     const std::string& name() const { return m_operator_node->name(); }
 
+    const std::string& params() const { return m_params; }
+
     const std::shared_ptr<mgb::ComputingGraph> get_owner_graph() const {
         return m_operator_node->owner_graph()->shared_from_this();
     }

python_module/src/swig/operator.i

@@ -58,6 +58,10 @@ public:
             return $self->name();
         }
 
+        const std::string& _get_params() const {
+            return $self->params();
+        }
+
         SymbolVarArray _get_inputs() {
             return $self->inputs();
         }

python_module/src/swig/operator.py

@@ -20,6 +20,11 @@ def name(self):
     return self._get_name()
 
 @property
+def params(self):
+    import json
+    return json.loads(self._get_params())
+
+@property
 def inputs(self):
     return tuple(self._get_inputs())