[XLA] Improve GPU memory limit handling and shape size calculation #23271
+186
−41
Google CLA / cla/google
succeeded
Feb 28, 2025 in 2s
✅ All contributors are covered under a CLA with Google
See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).
ℹ️ Googlers: Go here to view more details and manage scans for this pull request.
Details
The following contributors were found for this pull request:
✅ 8b31840 Author: @zhenying-liu <ja****u@nvidia.com>
(Only the first commit for a unique contributor is listed.)
Loading