Fix apache#17267, add expected and got datatype when non-uniform dtyp…

…e fount in concat
litaotju · Jan 11, 2020 · 0dc6f20 · 0dc6f20
1 parent 8c5990d
commit 0dc6f20
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/operator/nn/concat.cc b/src/operator/nn/concat.cc
@@ -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);
     }
   }