code: def device_idx(self) -> int:
      # TODO(PVP): currently only single GPU is supported
      return torch.cuda.current_device()
  
file_path: H2O/h2o_flexgen/benchmark/third_party/transformers/src/transformers/benchmark/benchmark_args.py
imports: from dataclasses import dataclass, field
  from typing import Tuple
  from ..utils import cached_property, is_torch_available, is_torch_tpu_available, logging, torch_required
  from .benchmark_args_utils import BenchmarkArguments
label_desc: Use image node_id 4 for calling the PyTorchBenchmarkArguments obj's underlying member method code with example usage: obj.device_idx() and returns: torch