Function _cuda_device_of

bitsandbytes/functional.py:82–83

(a: torch.Tensor)

if torch.cuda.device_count() > 1:

    def _cuda_device_of(a: torch.Tensor):
        return torch.cuda.device_of(a)
else:
    import contextlib