Add batch dim idx to support latest deepspeed DistributedAttention #1725

Merged

regisss merged 3 commits intohuggingface:synapse_1_20from HabanaAI:auto-pr-3f2094f

Feb 6, 2025

+76-10

Commits on Jan 8, 2025

[SW-207148] Add batch dim idx to support latest deepspeed DistributedAttention (#37 )

bhargaveede
authored and
astachowiczhabana
committed

Commits on Jan 27, 2025

Restructuring code (#100 )
bhargaveede
committed

Commits on Feb 6, 2025

Addressing review comments
bhargaveede
committed