Skip to content

Commit

Permalink
Fix apache#17267, add expected and got datatype when non-uniform dtyp…
Browse files Browse the repository at this point in the history
…e fount in concat
  • Loading branch information
litaotju committed Jan 11, 2020
1 parent 8c5990d commit 8bf1ad2
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions src/operator/nn/concat.cc
Original file line number Diff line number Diff line change
Expand Up @@ -149,9 +149,10 @@ bool ConcatType(const nnvm::NodeAttrs& attrs,
if (dtype == -1) {
dtype = i;
} else {
CHECK(i == dtype ||
i == -1) <<
"Non-uniform data type in Concat";
CHECK( i == dtype || i == -1)
<< "Non-uniform data type in Concat: "
<< "expected data type " << mxnet::op::type_string(dtype)
<< ", got data type " << mxnet::op::type_string(i);
}
}

Expand Down

0 comments on commit 8bf1ad2

Please sign in to comment.