From c9d984b377bc421b86472d66b08a676b504ad78a Mon Sep 17 00:00:00 2001 From: TianYu GUO Date: Wed, 27 Dec 2023 14:08:42 +0800 Subject: [PATCH 1/2] Fix minor error in doc --- docs/dev/how_to/relay_add_op.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/dev/how_to/relay_add_op.rst b/docs/dev/how_to/relay_add_op.rst index 225c60071fd0..25aecfa555ff 100644 --- a/docs/dev/how_to/relay_add_op.rst +++ b/docs/dev/how_to/relay_add_op.rst @@ -336,7 +336,7 @@ the arguments to the call node, as below. In ``src/relay/op/tensor/transform.cc` return Call(op, {data}, Attrs(attrs), {}); } - TVM_REGISTER_GLOBAL("relay.op._make.cumsum").set_body_typed(MakeCumprod); + TVM_REGISTER_GLOBAL("relay.op._make.cumprod").set_body_typed(MakeCumprod); Where ``TVM_REGISTER_GLOBAL`` exposes the ``MakeCumsum`` and ``MakeCumprod`` functions in Python via ``relay.op._make.cumsum(...)`` and ``relay.op._make.cumsum(...)``. From ef4192b55d8766c0dfe1762445c1b240ea8058e1 Mon Sep 17 00:00:00 2001 From: TianYu GUO Date: Wed, 27 Dec 2023 15:53:50 +0800 Subject: [PATCH 2/2] Update relay_add_op.rst --- docs/dev/how_to/relay_add_op.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/dev/how_to/relay_add_op.rst b/docs/dev/how_to/relay_add_op.rst index 25aecfa555ff..df0a4f7286f4 100644 --- a/docs/dev/how_to/relay_add_op.rst +++ b/docs/dev/how_to/relay_add_op.rst @@ -339,7 +339,7 @@ the arguments to the call node, as below. In ``src/relay/op/tensor/transform.cc` TVM_REGISTER_GLOBAL("relay.op._make.cumprod").set_body_typed(MakeCumprod); Where ``TVM_REGISTER_GLOBAL`` exposes the ``MakeCumsum`` and ``MakeCumprod`` functions -in Python via ``relay.op._make.cumsum(...)`` and ``relay.op._make.cumsum(...)``. +in Python via ``relay.op._make.cumsum(...)`` and ``relay.op._make.cumprod(...)``. 7. Including a Cleaner Python API Hook --------------------------------------