New Issues

Skip to first unread message

Patrick Toulme

Jun 2, 2024, 4:54:21 PMJun 2
to OpenXLA Discuss
I have opened four issues today for some bugs I have found in the past couple months for discussion. 

1. SPMD Partitioner Gather/Scatter is hardcoded to IndexParallel Strategy -
2. AllGather Combiner Coalesces different dtypes together -
3. SPMD Partitioner HandleReshape Unoptimally Splits Sharding in some cases - mainly in gradient accumulation case-
4. xla::gpu::ReduceScatterCreator misses one pattern -

I have fixes for these locally in my local fork, and I can try to upstream them unless others have easier solution. I also will be making PRs for some HLO passes I have written locally. 

Patrick Toulme

Amazon Web Services

George Karpenkov

Jun 3, 2024, 4:36:42 AMJun 3
to Patrick Toulme, Frederik Gossen, Greg Olechwierowicz, OpenXLA Discuss
NOTE: External mailing list.

+Frederik Gossen +Greg Olechwierowicz 

Makes sense to me, thanks!

You received this message because you are subscribed to the Google Groups "OpenXLA Discuss" group.
To unsubscribe from this group and stop receiving emails from it, send an email to
To view this discussion on the web visit
For more options, visit
Reply all
Reply to author
0 new messages