In the topi, we could get the target information during schedule using
`tvm.target.Target.current()`. But we don't have `target_host` information as
far as I know.
But seems that you could do in `def _build_for_device` (we could add pass
inside it like other passes). However, you should double check we maybe go to
this function if we use relay build (pure C++ way) and doesn't enter into `def
_build_for_device`:
```
runtime::Module build(const Map<Target, IRModule>& inputs,
const Target& target_host,
const BuildConfig& config)
```
I haven't verified it.
---
[Visit Topic](https://discuss.tvm.ai/t/remove-an-attribute-from-ir/6526/6) to
respond.
You are receiving this because you enabled mailing list mode.
To unsubscribe from these emails, [click
here](https://discuss.tvm.ai/email/unsubscribe/e4f8f005d2483a9aff295752d62d7d9e9f56375583a763153a630d78c3a1e121).
