Skip to content

Commit 04ccfef

Browse files
committed
Fixed failing test in unity CI
1 parent 8f39394 commit 04ccfef

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/tir/transforms/annotate_device_regions.cc

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -131,6 +131,10 @@ class DeviceRegionAnnotater : public StmtExprMutator {
131131
op->op.same_as(builtin::tvm_call_packed()) || op->op.same_as(builtin::tvm_call_cpacked()) ||
132132
op->op.same_as(builtin::tvm_call_packed_lowered()) ||
133133
op->op.same_as(builtin::tvm_call_cpacked_lowered()) ||
134+
op->op.same_as(builtin::anylist_getitem()) ||
135+
op->op.same_as(builtin::anylist_resetitem()) ||
136+
op->op.same_as(builtin::anylist_setitem_call_packed()) ||
137+
op->op.same_as(builtin::anylist_setitem_call_cpacked()) ||
134138
op->op.same_as(builtin::tvm_struct_get()) || op->op.same_as(builtin::tvm_struct_set()) ||
135139
op->op.same_as(builtin::tvm_throw_last_error()) ||
136140
op->op.same_as(builtin::tvm_stack_alloca()) ||

0 commit comments

Comments
 (0)