Browse Source

fix bug int32 to int64

tags/v1.1.0
simson 5 years ago
parent
commit
0c52d69ffa
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/transform/graph_ir/convert.cc

+ 1
- 1
mindspore/ccsrc/transform/graph_ir/convert.cc View File

@@ -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;


Loading…
Cancel
Save