From 9dff320d9cc3fb10f43007e554cf1b05cf70b925 Mon Sep 17 00:00:00 2001 From: fary86 Date: Mon, 7 Sep 2020 20:01:56 +0800 Subject: [PATCH] Fix SparseApplyFtrl ut segmentfault in pynative mode --- mindspore/core/abstract/abstract_value.h | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/mindspore/core/abstract/abstract_value.h b/mindspore/core/abstract/abstract_value.h index f9bf8f84ba..b66e5a9c3f 100644 --- a/mindspore/core/abstract/abstract_value.h +++ b/mindspore/core/abstract/abstract_value.h @@ -563,7 +563,9 @@ class AbstractRefKey : public AbstractBase { } inline void set_value(const ValuePtr &value) { AbstractBase::set_value(value); - ref_key_value_ = value->cast(); + if (value != nullptr) { + ref_key_value_ = value->cast(); + } } RefKeyPtr ref_key_value() const { return ref_key_value_; } AbstractBasePtr Join(const AbstractBasePtr &other) override;