Skip to content

Commit 85b1d5e

Browse files
committed
[EthosU] Use ethos-u as both device and host
1 parent e9a4c32 commit 85b1d5e

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

python/tvm/relay/backend/contrib/ethosu/tir/compiler.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -209,7 +209,9 @@ def transform_npu_function(self, _, func: relay.Function) -> relay.Function:
209209
primfunc = tir_mod["main"]
210210
primfunc = primfunc.with_attr("global_symbol", func.attrs["global_symbol"])
211211
primfunc = primfunc.with_attr("ethos-u.constants", const_dict)
212-
primfunc = primfunc.with_attr("target", tvm.target.Target(compiler_name))
212+
primfunc = primfunc.with_attr(
213+
"target", tvm.target.Target(compiler_name, host=compiler_name)
214+
)
213215
return primfunc
214216

215217
def __call__(self, *args, **kwargs):

0 commit comments

Comments
 (0)