[RFC] Add source_target_pairs to send and recv ops in stableHLO

27 views
Skip to first unread message

Rosie Zou

unread,
Apr 23, 2025, 1:44:02 PMApr 23
to OpenXLA Discuss
Hi folks, PTAL at my pull request here https://github.com/openxla/stablehlo/pull/2784. This RFC adds the source_target_pairs attribute to send/recv operations in stableHLO and is a part of a larger workflow to expose send/recv API to JAX users. Feedback is greatly appreciated!
Reply all
Reply to author
Forward
0 new messages