Skip to content

Commit e9a4c32

Browse files
committed
Allow call_extern in device-side functions
May be used by kernels to call device-specific intrinsics (e.g. for cmsis-nn)
1 parent 358d459 commit e9a4c32

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

src/tir/transforms/annotate_device_regions.cc

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,6 @@ class DeviceRegionAnnotater : public StmtExprMutator {
132132
op->op.same_as(builtin::tvm_call_packed_lowered()) ||
133133
op->op.same_as(builtin::tvm_call_cpacked_lowered()) ||
134134
op->op.same_as(builtin::tvm_struct_get()) || op->op.same_as(builtin::tvm_struct_set()) ||
135-
op->op.same_as(builtin::call_extern()) || op->op.same_as(builtin::call_pure_extern()) ||
136135
op->op.same_as(builtin::tvm_throw_last_error()) ||
137136
op->op.same_as(builtin::tvm_stack_alloca()) ||
138137
op->op.same_as(builtin::tvm_stack_make_shape()) ||

0 commit comments

Comments
 (0)