Allow call_extern in device-side functions

May be used by kernels to call device-specific intrinsics (e.g. for cmsis-nn)
apache · Jul 12, 2023 · eb8bd78 · eb8bd78
1 parent ddae3ad
commit eb8bd78
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/tir/transforms/annotate_device_regions.cc b/src/tir/transforms/annotate_device_regions.cc
@@ -132,7 +132,6 @@ class DeviceRegionAnnotater : public StmtExprMutator {
         op->op.same_as(builtin::tvm_call_packed_lowered()) ||
         op->op.same_as(builtin::tvm_call_cpacked_lowered()) ||
         op->op.same_as(builtin::tvm_struct_get()) || op->op.same_as(builtin::tvm_struct_set()) ||
-        op->op.same_as(builtin::call_extern()) || op->op.same_as(builtin::call_pure_extern()) ||
         op->op.same_as(builtin::tvm_throw_last_error()) ||
         op->op.same_as(builtin::tvm_stack_alloca()) ||
         op->op.same_as(builtin::tvm_stack_make_shape()) ||