[feature request] hash of torch.Device #7228
Description
Do you think it is a good idea that torch.Device support hashing ?
hash(torch.device("cpu"))
TypeError: unhashable type: 'torch.Device'On my side, it would be useful for cached functions like
@lru_cache()
def big_constant_tensor(x, y, device):
return torch.tensor((1000, 1000), math.pi, device=device)Currently I do this to avoid the problem
@lru_cache()
def big_constant_tensor(x, y, device_type, device_index):
return torch.tensor((1000, 1000), math.pi, device=torch.device(device_type, device_index))