From b3ff7c47dd9627fa55736ec86abdafed88918eb9 Mon Sep 17 00:00:00 2001 From: Eric Lunderberg Date: Wed, 8 May 2024 08:18:02 -0500 Subject: [PATCH] Fixed failing test in unity CI --- src/tir/transforms/annotate_device_regions.cc | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/tir/transforms/annotate_device_regions.cc b/src/tir/transforms/annotate_device_regions.cc index 4c5209e4b3b35..67980a934219e 100644 --- a/src/tir/transforms/annotate_device_regions.cc +++ b/src/tir/transforms/annotate_device_regions.cc @@ -131,6 +131,10 @@ class DeviceRegionAnnotater : public StmtExprMutator { op->op.same_as(builtin::tvm_call_packed()) || op->op.same_as(builtin::tvm_call_cpacked()) || op->op.same_as(builtin::tvm_call_packed_lowered()) || op->op.same_as(builtin::tvm_call_cpacked_lowered()) || + op->op.same_as(builtin::anylist_getitem()) || + op->op.same_as(builtin::anylist_resetitem()) || + op->op.same_as(builtin::anylist_setitem_call_packed()) || + op->op.same_as(builtin::anylist_setitem_call_cpacked()) || op->op.same_as(builtin::tvm_struct_get()) || op->op.same_as(builtin::tvm_struct_set()) || op->op.same_as(builtin::tvm_throw_last_error()) || op->op.same_as(builtin::tvm_stack_alloca()) ||