Add Primitive::name and remove Primitive::print (#2365)

2025-09-01 12:49:44 +08:00 · 2025-07-15 06:06:35 +09:00
parent 5201df5030
commit d34f887abc
32 changed files with 307 additions and 340 deletions
--- a/examples/extensions/axpby/axpby.h
+++ b/examples/extensions/axpby/axpby.h
@@ -74,9 +74,9 @@ class Axpby : public mx::Primitive {
      const std::vector<mx::array>& inputs,
      const std::vector<int>& axes) override;

-  /** Print the primitive. */
-  void print(std::ostream& os) override {
-    os << "Axpby";
+  /** The name of primitive. */
+  const char* name() const override {
+    return "Axpby";
  }

  /** Equivalence check **/