Error msg update (#5818)
authorANSHUMAN TRIPATHY <anshuman.t@huawei.com>
Tue, 16 Jun 2020 15:24:12 +0000 (20:54 +0530)
committerGitHub <noreply@github.com>
Tue, 16 Jun 2020 15:24:12 +0000 (08:24 -0700)
include/tvm/node/container.h
src/ir/expr.cc

index a3cfdaf..fa8c875 100644 (file)
@@ -279,7 +279,7 @@ struct ObjectTypeChecker<Array<T> > {
     }
     return true;
   }
-  static std::string TypeName() { return "List[" + ObjectTypeChecker<T>::TypeName() + "]"; }
+  static std::string TypeName() { return "Array[" + ObjectTypeChecker<T>::TypeName() + "]"; }
 };
 
 template <typename K, typename V>
index 289477e..b032fe5 100644 (file)
@@ -57,7 +57,7 @@ PrimExpr PrimExpr::FromObject_(ObjectRef ref) {
 
 IntImm::IntImm(DataType dtype, int64_t value) {
   CHECK(dtype.is_scalar()) << "ValueError: IntImm can only take scalar.";
-  CHECK(dtype.is_int() || dtype.is_uint()) << "ValueError: IntImm can only take scalar.";
+  CHECK(dtype.is_int() || dtype.is_uint()) << "ValueError: IntImm supports only int or uint type.";
   if (dtype.is_uint()) {
     CHECK_GE(value, 0U);
   }