10000 Create a new tensor arg for stride_per_key_per_rank to facilitate torch.export by jd7-tr · Pull Request #2950 · pytorch/torchrec · GitHub
[go: up one dir, main page]
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 intopytorch: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 intopytorch:mainfrom
jd7-tr:export-D74207283

Commits

Commits on May 7, 2025
0