Add str type to device parameter of torch.cuda.get_device_name() on the doc

[hyperkai]

📚 The doc issue

The doc explains device parameter but it doesn't have str type as shown below:

device (torch.device or int, optional) – device for which to return the name. This function is a no-op if this argument is a negative integer. It uses the current device, given by current_device(), if device is None (default).

Actually, device parameter with str type works as shown below:

import torch

torch.cuda.get_device_name(device='cuda:0') # Tesla T4

Suggest a potential alternative/fix

The doc should also say str type as shown below:

device (torch.device, int or str, optional) – device for which to return ...

cc @ptrblck @msaroufim @albanD

[shink]

this case works because the string you passed in will be used to create an torch.device object, I guess that there are a lot of function signatures that look like this, so I think it's unnecessary to update the doc and any suggestions are welcome

pytorch/c10/core/Device.h

Lines 41 to 46 in 6c503f1

	/// Constructs a `Device` from a string description, for convenience.
	/// The string supplied must follow the following schema:
	/// `(cpu|cuda)[:<device-index>]`
	/// where `cpu` or `cuda` specifies the device type, and
	/// `:<device-index>` optionally specifies a device index.
	/* implicit */ Device(const std::string& device_string);

[albanD]

Yes updating the doc sounds good.
If you need typing, you can use

pytorch/torch/types.py

Line 39 in 2f53747

Device = Optional[Union[_device, str, _int]]