unbox expects Dml at::Tensor as inputs

microsoft / tensorflow-directml

Fork of TensorFlow accelerated by DirectML

Apache License 2.0

457 stars 32 forks source link

unbox expects Dml at::Tensor as inputs #404

Closed mengdalida closed 1 year ago

mengdalida commented 1 year ago

RuntimeError: tensor.device().type() == at::DeviceType::PrivateUse1 INTERNAL ASSERT FAILED at "D:\a\_work\2\s\pytorch-directml-plugin\torch_directml\csrc\dml\DMLTensor.cpp":31, please report a bug to PyTorch. unbox expects Dml at::Tensor as inputs

my code is

s1 = einsum('b i d, b j d -> b i j', q[:, i:end].cpu(), k.cpu())

maggie1059 commented 1 year ago

Hi @mengdalida, it looks like you're using the torch-directml plugin. For PyTorch-DirectML issues, please open an issue on the DirectML GitHub.