diff --git a/src/tirx/analysis/verify_memory.cc b/src/tirx/analysis/verify_memory.cc index a6c3c0ef3552..d97be6c5daae 100644 --- a/src/tirx/analysis/verify_memory.cc +++ b/src/tirx/analysis/verify_memory.cc @@ -168,7 +168,9 @@ class MemoryAccessVerifier final : protected StmtExprVisitor { /// Interface of VerifyMemory pass std::vector VerifyMemory_(const PrimFunc& func) { auto target = func->GetAttr(tvm::attr::kTarget); - TVM_FFI_ICHECK(target.defined()) << "VerifyMemory: Require the target attribute"; + // Skip verification for functions without a target attribute, as they are + // typically host-only helper functions that do not have device-memory constraints. + if (!target.defined()) return {}; VLOG(1) << "verifying memory for target '" << target.value()->str() << "' for primitive:" << std::endl