If you are using `relay.build()` -> `graph_executor.GraphModule` path, the
point I remember is that it should pass a multi-target dict into `target`
argument of build and pass a device list into GraphModule like
```python
lib = relay.build(relay_mod, target={"cpu": "llvm", "gpu": "cuda"},
params=params)
m = graph_executor.GraphModule(lib["default"](tvm.cpu(), tvm.gpu()))
```
---
[Visit
Topic](https://discuss.tvm.apache.org/t/how-to-do-heterogeneous-execution-on-cpu-and-gpu/11561/3)
to respond.
You are receiving this because you enabled mailing list mode.
To unsubscribe from these emails, [click
here](https://discuss.tvm.apache.org/email/unsubscribe/0e823c9c4118e3d4b9c4de99d3b6f2e78c4d7b44a9fed14480b0664bd09f63e6).
