[sharding_in_types] If out_aval.sharding is not None and the user specified out_sharding is None, concretize it with the device assignment available and add it to the final out_shardings that's used for lowering and compilation. #24387
+43
−26
Google CLA / cla/google
succeeded
Oct 18, 2024 in 1s
✅ 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:
✅ 2153de4 PR Opener: @copybara-service[bot]
✅ 2153de4 Author: @yashk2810 <yas******iya@google.com>
(Only the first commit for a unique contributor is listed.)
Loading