compatible with DirectML

#71
by davinwang - opened

Tensor.new is a deprecated constructor and does not support PrivateUse1 in pytorch 1.13.1/2.0.0, use torch.ones instead. Please refer to https://github.com/microsoft/DirectML/issues/400 and https://github.com/pytorch/pytorch/issues/95734

This should also fix the ROCm compatibility in this file

Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment