How to set default tensor device as HPU?

Training PyTorch
1

If we use cuda as pytorch device, we can use

torch.set_default_tensor_type(‘torch.cuda.FloatTensor’)

to set default tensor device as cuda when we create a tensor.
but what should I do if I want to create a tensor with default device as HPU?

2

Thanks for posting, will check and get back to you

3

You can create tensors on device by:
x = torch.tensor([1.0,2.0], device='hpu')
or move them to device by:

x = torch.tensor([1.0,2.0])
x = x.to('hpu')

However you cannot set a default.