Commit 44c0b5a
committed
Update base for Update on "[ET Device Support] Propagate device info from TensorSpec into serialized Tensor"
Propagate device information from `TensorSpec.device` (set by `PropagateDevicePass`) to
the serialized `schema.Tensor` in the emitted PTE file, to make runtime further aware of it.
Differential Revision: [D95899706](https://our.internmc.facebook.com/intern/diff/D95899706/)
[ghstack-poisoned]1 parent 46fd990 commit 44c0b5a
1 file changed
Lines changed: 5 additions & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
144 | 144 | | |
145 | 145 | | |
146 | 146 | | |
147 | | - | |
| 147 | + | |
| 148 | + | |
| 149 | + | |
| 150 | + | |
| 151 | + | |
148 | 152 | | |
149 | 153 | | |
150 | 154 | | |
| |||
0 commit comments