Skip to content

Create a new tensor arg for stride_per_key_per_rank to facilitate torch.export#2950

Open
jd7-tr wants to merge 2 commits intometa-pytorch:mainfrom
jd7-tr:export-D74207283
Open

Create a new tensor arg for stride_per_key_per_rank to facilitate torch.export#2950
jd7-tr wants to merge 2 commits intometa-pytorch:mainfrom
jd7-tr:export-D74207283

Commits

Commits on May 7, 2025