NVIDIA / NVIDIA/Model-Optimizer
ERROR：train sparse model with FDSP2
Open
@kaix-nv is already working on this.
Since Jan 29, 2026.
bug
torch.sparsity
- Dominant language
- Python
- Stars
- 3.8k
- Forks
- 604
- Avg merge
- 2d 8h
- Merged PRs (30d)
- 142
Description
Bug:
How to solve it?
masked_weight = weight * mod._weight_mask
RuntimeError: aten.mul.Tensor: got mixed torch.Tensor and DTensor, need to convert all torch.Tensor to DTensor before calling distributed operators!
system info:
torch 2.9.0+cu12.6
modelopt 0.41
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.