DeepSpeed
177dc143 - Fix UserWarning: The torch.cuda.*DtypeTensor constructors are no long… (#5018)

Commit
1 year ago
Fix UserWarning: The torch.cuda.*DtypeTensor constructors are no long… (#5018) > …er recommended. It's best to use methods such as torch.tensor(data, dtype=*, device='cuda') to create tensors. By currying the returned tensor properties in CUDA_Accelerator, the above warning in PyTorch is avoided. Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com> Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
Author
Parents
Loading