Fix SparseApplyFtrl ut segmentfault in pynative mode

5 years ago · 9dff320d9c
parent 72f77bdea7
commit 9dff320d9c
1 changed files with 3 additions and 1 deletions
--- 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<RefKeyPtr>();
+    if (value != nullptr) {
+      ref_key_value_ = value->cast<RefKeyPtr>();
+    }
  }
  RefKeyPtr ref_key_value() const { return ref_key_value_; }
  AbstractBasePtr Join(const AbstractBasePtr &other) override;