Skip to content

Commit 4dd903d

Browse files
GasoonjiaGasoonjia
committed
Update base for Update on "[ET Device Support] TensorImpl carries device info"
This diff extends `TensorImpl` to carry device information, enabling the runtime tensor to track which device its data resides on (CPU, CUDA, etc.). This is a prerequisite for parsing device info from the schema and allocating device memory. Differential Revision: [D93635655](https://our.internmc.facebook.com/intern/diff/D93635655/) [ghstack-poisoned]
1 parent 33acf17 commit 4dd903d

0 file changed

File tree

    0 commit comments

    Comments
     (0)