- bool is legit IntImm

apache · May 6, 2022 · 088671f · 088671f
1 parent c655b0f
commit 088671f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/support/scalars.cc b/src/support/scalars.cc
@@ -130,6 +130,8 @@ std::string IntImmToString(const IntImm& int_imm) {
     os << int_imm->value;
   } else if (int_imm->dtype == kInt64) {
     os << int_imm->value << "i64";
+  } else if (int_imm->dtype == kBool) {
+    os << (int_imm->value ? "True" : "False");
   } else {
     LOG(FATAL) << "Unrecognised IntImm dtype: " << DLDataType2String(int_imm->dtype);
   }
@@ -150,8 +152,6 @@ std::string FloatImmToString(const FloatImm& float_imm) {
   return os.str();
 }
 
-std::string BoolToString(bool value) { return value ? "True" : "False"; }
-
 std::pair<IntImm, bool> ValueToIntImm(int64_t value, int width) {
   bool clipped = false;
   if (width == 16) {

diff --git a/src/support/scalars.h b/src/support/scalars.h
@@ -52,7 +52,6 @@ std::string NDArrayScalarToString(const runtime::NDArray& data);
 /*! \brief Returns Relay literal text for given TIR immediate. */
 std::string IntImmToString(const IntImm& int_imm);
 std::string FloatImmToString(const FloatImm& float_imm);
-std::string BoolToString(bool value);
 
 /*!
  * \brief Returns TIR immediate for given value and width. Boolean will be true if value