Add physical device ordinal to buffers (TrackedDeviceBuffer, ShapedBuffer). - Githubissues

openxla / xla

A machine learning compiler for GPUs, CPUs, and ML accelerators

Apache License 2.0

2.39k stars 358 forks source link

Add physical device ordinal to buffers (TrackedDeviceBuffer, ShapedBuffer). #14229

Open copybara-service[bot] opened 6 days ago

copybara-service[bot] commented 6 days ago

Add physical device ordinal to buffers (TrackedDeviceBuffer, ShapedBuffer).