Manually migrate bd changes to 2.18 branch

ROCm · Dec 6, 2024 · e3328fe · e3328fe
1 parent 432b95b
commit e3328fe
Show file tree

Hide file tree

Showing 3 changed files with 678 additions and 234 deletions.
diff --git a/tensorflow/core/framework/device_base.h b/tensorflow/core/framework/device_base.h
@@ -165,6 +165,8 @@ class DeviceBase {
     int gpu_id = -1;
   };
 
+  using GpuDeviceInfo = AcceleratorDeviceInfo;
+
   // Does not take ownership.
   void set_tensorflow_accelerator_device_info(
       AcceleratorDeviceInfo* device_info) {