Fixed accidently removed scope tag check

apache · Jul 28, 2021 · 9398d96 · 9398d96
1 parent a290195
commit 9398d96
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/tir/transforms/lower_device_storage_access_info.cc b/src/tir/transforms/lower_device_storage_access_info.cc
@@ -42,7 +42,7 @@ class StorageAccessInfoLower : public StmtExprMutator {
  public:
   Stmt VisitStmt_(const AllocateNode* op) final {
     auto scope = StorageScope::Create(GetPtrStorageScope(op->buffer_var));
-    if (scope.tag.length() != 0) {
+    if (scope.tag.length() != 0 && scope.tag != ".dyn") {
       auto info = GetMemoryInfo(GetPtrStorageScope(op->buffer_var));
       ICHECK(info.defined()) << "Cannot find memory info of " << scope.to_string();
       ICHECK(storage_info_.find(op->buffer_var.get()) == storage_info_.end())