提交 528ce2b9 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!2767 fix hook operator compare issue

Merge pull request !2767 from wangqiuliang/fix-hook-operator-compare-issue
...@@ -106,28 +106,23 @@ FuncGraphPtr KPrim::KPrimitive(const ValueNodePtr &value_node, const pipeline::R ...@@ -106,28 +106,23 @@ FuncGraphPtr KPrim::KPrimitive(const ValueNodePtr &value_node, const pipeline::R
} }
FuncGraphPtr bprop_fg = nullptr; FuncGraphPtr bprop_fg = nullptr;
auto iter = bprop_registry_.find(prim); if (prim->Hash() == prim::kPrimHookBackward->Hash() && prim->name() == prim::kPrimHookBackward->name()) {
if (iter != bprop_registry_.end()) { bprop_fg = BpropCut(value_node, resources);
bprop_fg = iter->second; } else {
} auto iter = bprop_registry_.find(prim);
if (iter != bprop_registry_.end()) {
bprop_fg = iter->second;
}
if (bprop_fg == nullptr) { if (bprop_fg == nullptr) {
bool is_faked_bprop = false;
if (prim->Hash() == prim::kPrimHookBackward->Hash() && prim->name() == prim::kPrimHookBackward->name()) {
bprop_fg = BpropCut(value_node, resources);
} else {
bprop_fg = GetBprop(prim); bprop_fg = GetBprop(prim);
if (bprop_fg == nullptr) { if (bprop_fg != nullptr) {
// Set bprop_g graph cache
bprop_registry_[prim] = bprop_fg;
} else {
bprop_fg = FakeBprop(value_node, resources); bprop_fg = FakeBprop(value_node, resources);
is_faked_bprop = true;
} }
} }
// To support primitives with variable params, do not cache faked bprop
if (!is_faked_bprop) {
// Set bprop_g graph cache
bprop_registry_[prim] = bprop_fg;
}
} }
auto expanded_fg = BpropToK(prim, bprop_fg); auto expanded_fg = BpropToK(prim, bprop_fg);
......
...@@ -109,6 +109,9 @@ class Tensor(Tensor_): ...@@ -109,6 +109,9 @@ class Tensor(Tensor_):
out = tensor_operator_registry.get('__neg__')(self) out = tensor_operator_registry.get('__neg__')(self)
return out return out
def __pos__(self):
return self
def __iadd__(self, other): def __iadd__(self, other):
return self.__add__(other) return self.__add__(other)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册