Commit a3aa2ff
committed
Update base for Update on "[executorch] Propagate device metadata from partitioner result onto TensorSpecs"
Add end-to-end device type annotation support from export to runtime. Currently we only support one device per graph
The overall pipeline is:
a. Partitioner use `compile_spec` to determine which device the partitoned blob is runing on
b. after lowered partitioned graph to backend, the new-introed propagate_device_pass will annotate the input and output tensors of delegate blob as target device.
Differential Revision: [D95842511](https://our.internmc.facebook.com/intern/diff/D95842511/)
[ghstack-poisoned]1 parent 1fb6f12 commit a3aa2ff
0 file changed
0 commit comments