[PASS] Check memory info bound to guard failure (#409)

src/pass/storage_flatten.cc

@@ -140,6 +140,8 @@ class StorageFlattener : public IRMutator {
         MemoryInfo info = GetMemoryInfo(skey.to_string());
         if (info.defined()) {
           align = (info->max_simd_bits + op->type.bits() - 1) / op->type.bits();
+          CHECK_LE(const_size * op->type.bits(), info->max_num_bits)
+              << "Allocation exceed bound of memory tag " << skey.to_string();
         }
       }
       Array<Expr> strides;