Fix error reporting for missing axis (apache#2835)

* Fix error reporting for missing axis * Update data_layout.cc
wweic · Mar 24, 2019 · c4a63d1 · c4a63d1
1 parent 495b52c
commit c4a63d1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/lang/data_layout.cc b/src/lang/data_layout.cc
@@ -5,6 +5,7 @@
  */
 #include <tvm/data_layout.h>
 #include <tvm/ir_pass.h>
+#include <cctype>
 
 namespace tvm {
 
@@ -115,7 +116,7 @@ Layout::Layout(const std::string& name) { // NOLINT(*)
     char axis = v->var.get()->name_hint[0];
     if (axis >= 'a' && axis <= 'z') {
       CHECK(exist_axis[axis-'a'+'A']) << "Invalid layout " << name << ": missing axis "
-                                      << axis - 'a' + 'A';
+                                      << std::toupper(axis);
     }
   }
 }