New Issues

70 views
Skip to first unread message

Patrick Toulme

unread,
Jun 2, 2024, 4:54:21 PMJun 2
to OpenXLA Discuss
Hello, 
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 - https://github.com/openxla/xla/issues/13304
2. AllGather Combiner Coalesces different dtypes together - https://github.com/openxla/xla/issues/13305
3. SPMD Partitioner HandleReshape Unoptimally Splits Sharding in some cases - mainly in gradient accumulation case- https://github.com/openxla/xla/issues/13306
4. xla::gpu::ReduceScatterCreator misses one pattern - https://github.com/openxla/xla/issues/13307

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. 

Regards, 
Patrick Toulme

Amazon Web Services

George Karpenkov

unread,
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 openxla-discu...@openxla.org.
To view this discussion on the web visit https://groups.google.com/a/openxla.org/d/msgid/openxla-discuss/b4686b97-6456-49a2-a4e8-56c67992e6fan%40openxla.org.
For more options, visit https://groups.google.com/a/openxla.org/d/optout.
Reply all
Reply to author
Forward
0 new messages