viable/strict/1792049632: [TokenSwitch] Specify the backend contract and test it in CI (#198963)
- PyTorch: 2354 events in the last 90 days
- PyTorch: 2324th Release in the last 90 days
- Previous: earlier the same day · trunk/0849859f46843d5b076266b71f3ca4c3359522b8
What happened
We're adding a ROCm MoRI backend for TokenSwitch. While working on the implementation, it baked some implicit semantics from the NCCL EP behavior into TokenSwitchNCCL. This PR establishes the API and behavior contract on its own, with new CI coverage, that will pave the way for the new TokenSwitchMoRI implementation. Claude (AI assistant) drafted the following description of the change: Why. TokenSwitch's semantics…
Summary assembled by rule from the sources below