DeepSpeed
8d1bc0a0 - Update `torch.norm` to `torch.linalg.norm` and `torch.linalg.vector_norm` (#6931)

Commit
245 days ago
Update `torch.norm` to `torch.linalg.norm` and `torch.linalg.vector_norm` (#6931) - [x] Update PR since `torch.norm` and `torch.linalg.norm` have [different function signatures](https://pytorch.org/docs/stable/generated/torch.linalg.norm.html#torch.linalg.norm). - [x] Check if there are any numeric differences between the functions. - [x] Determine why there appear to be performance differences from others [here](https://github.com/pytorch/pytorch/issues/136360). - [x] Update to `torch.linalg.vectornorm` Follow up PR handles these in the comm folder: #6960
Author
Parents
Loading