[Feat] 算子注册

[aztice]

PR For

Cpp

PR Types

Feat

Description

// PyBind中算子部分不再通过手动 bind 实现绑定，而是算作是算子形式
axono::core::OpRegistry::instance().bind_all(m);

注册方法

// 1. 定义一个 py::object
py::object op_impl_$算子名(const py::args& args);
// 2. 开始注册
REGISTER_OP($算子名) {
  // 3. 可选（参数筛选）
  if (args.size() != 2) {
    throw std::runtime_error("执行 $参数名 需要传入 2 个 Tensor 喵~");
  }
  auto& a = pybind11::cast<core::Tensor&>(args[0]);
  auto& b = pybind11::cast<core::Tensor&>(args[1]);
  // 4. 你的实现（建议遵循 Axono 通用规范，也可自由发挥如是个人实验用途）
  // ...

  // 5. 返回转换
  return pybind11::cast($result);
}

[paddle-bot]

Thanks for your contribution!

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.

📢 Thoughts on this report? Let us know!