From 9398d96f2d9d3c63dfd81fc37195092d727a8bac Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Tue, 27 Jul 2021 23:43:55 +0900 Subject: [PATCH] Fixed accidently removed scope tag check --- src/tir/transforms/lower_device_storage_access_info.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/tir/transforms/lower_device_storage_access_info.cc b/src/tir/transforms/lower_device_storage_access_info.cc index 7443a2449fb2..b4ec91ba5012 100644 --- 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())