fix bug int32 to int64

4 years ago · 0c52d69ffa
parent 01f5f6a060
commit 0c52d69ffa
1 changed files with 1 additions and 1 deletions
--- a/mindspore/ccsrc/transform/graph_ir/convert.cc
+++ b/mindspore/ccsrc/transform/graph_ir/convert.cc
@ -1426,7 +1426,7 @@ bool DfGraphConvertor::GetControlDependList(const CNodePtr &node,
  ValuePtr mode_ptr = prim_ptr->GetAttr("depend_mode");
  int depend_mode = DEPEND_MODE_NORMAL_USE;
  if (mode_ptr != nullptr) {
-    auto mode_int = mode_ptr->cast<Int32ImmPtr>();
+    auto mode_int = mode_ptr->cast<Int64ImmPtr>();
    MS_EXCEPTION_IF_NULL(mode_int);
    depend_mode = mode_int->value();
    MS_LOG(DEBUG) << "depend_mode = " << depend_mode;