From b98a07bd382203e76b0e5c67108bfdad023d35c8 Mon Sep 17 00:00:00 2001 From: Tao Li Date: Sat, 11 Jan 2020 14:02:40 +0800 Subject: [PATCH] Fix #17267, add expected and got datatype when non-uniform dtype fount in concat --- src/operator/nn/concat.cc | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/operator/nn/concat.cc b/src/operator/nn/concat.cc index 1eeef7db5cb5..663bbe1e933e 100644 --- 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); } }