xla
xla copied to clipboard
Published
20 hours ago •
openxla
openxla
[xla:spmd] Fix a problem in generating code for concat that requires padding.
trafficstars
[xla:spmd] Fix a problem in generating code for concat that requires padding.
When an internal concat operand requires padding, previously, we don't zero out the padded elements and the generated code produces incorrect results.
Fix two existing tests.
![copybara-service[bot] avatar](https://avatars.githubusercontent.com/in/44061?v=4 "Published by a pub.dev verified publisher"){kind=small}