Update base for Update on "[ET Device Support] TensorImpl carries device info"

Gasoonjia · Gasoonjia · commit c1d5e1a563a4 · 2026-03-19T13:55:55.000-07:00
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]
diff --git a/schema/program.fbs b/schema/program.fbs
@@ -63,6 +63,7 @@ enum TensorDataLocation : byte {
 }
 
 // Device type enum indicating where a tensor resides or should be allocated.
+// Please keep this in sync with executorch/runtime/core/portable_type/device.h
 enum DeviceType : byte {
   CPU = 0,
   CUDA = 1,

Original file line number	Diff line number	Diff line change
`@@ -63,6 +63,7 @@ enum TensorDataLocation : byte {`
`63`	`63`	`}`
`64`	`64`
`65`	`65`	`// Device type enum indicating where a tensor resides or should be allocated.`
	`66`	`+// Please keep this in sync with executorch/runtime/core/portable_type/device.h`
`66`	`67`	`enum DeviceType : byte {`
`67`	`68`	`CPU = 0,`
`68`	`69`	`CUDA = 1,`