Fixed failing test in unity CI

apache · May 8, 2024 · b3ff7c4 · b3ff7c4
1 parent 25a5de0
commit b3ff7c4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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()) ||