[PyTorch] Batch CP attention tests in single torchrun to amortize NCC…#2965
Open
sudhakarsingh27 wants to merge 2 commits intoNVIDIA:mainfrom
Open
[PyTorch] Batch CP attention tests in single torchrun to amortize NCC…#2965sudhakarsingh27 wants to merge 2 commits intoNVIDIA:mainfrom
sudhakarsingh27 wants to merge 2 commits intoNVIDIA:mainfrom
![pre-commit-ci[bot]](https://avatars.githubusercontent.com/in/68672?v=4&size=32){kind=avatar}