trunk/542330ebe79bfc3e9275d7610f9adcf489e3abf9: Fix scaled-MM DeepSeek indexing and contraction validation (#197028)
- PyTorch: 1027 events in the last 90 days
- PyTorch: 1016th Release in the last 90 days
- Previous: earlier the same day · trunk/7083f3a16277e0e4f58b5064dcc5d9840bf2c24c: [cuda][nccl] Drop the dead `NCCL_ALLTOALL_SUPPORTED` arm (#196400)
What happened
Human note I found this repeatedly failing on blackwell CI Agent note The DeepSeek Triton main loop assumed row-major scale storage, advanced scale indices incorrectly for GEMM tiles smaller than a scale block, and interleaved outer/K dimensions when expanding 128x128 scales. Honor actual strides and logical axes, compute scale coordinates from element offsets, and preserve outer/K order during expansion. Explicit v…
Summary assembled by rule from the sources below