Add support for `torch.set_default_device` when initializing model parameters - #1158

Open

fshp971 wants to merge 1 commit into

facebookresearch:mainfrom

fshp971:create-parameter-via-torch-empty

Commits on Dec 28, 2023

Replace `torch.Tensor` with `torch.empty` in `fairscale/nn/model_parallel/layers.py` to support `torch.set_default_device`
fshp971
committed
37d0a0d