[TIR] Do not add kIsHostFunc attribute during SplitHostDevice

Per discussion on apache#14020, the `kIsHostFunct` attribute should only be used in `BindTarget`, and should not be re-introduced in `SplitHostDevice`.
Lunderberg · May 25, 2023 · a8d7ca6 · a8d7ca6
1 parent 5e9fac8
commit a8d7ca6
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/tir/transforms/split_host_device.cc b/src/tir/transforms/split_host_device.cc
@@ -116,7 +116,6 @@ PrimFunc SplitHostDevice(PrimFunc func, IRModule* device_mod, const GlobalVar& g
   auto body = splitter(func->body);
 
   func.CopyOnWrite()->body = body;
-  func = WithAttr(std::move(func), tvm::tir::attr::kIsHostFunc, Bool(true));
 
   return func;
 }