code: def gpu_device(i):
      device_option = caffe2_pb2.DeviceOption()
      device_option.device_type = workspace.GpuDeviceType
      device_option.device_id = i
      return device_option
  
file_path: pytorch/caffe2/contrib/nccl/nccl_ops_test.py
imports: import unittest
  import hypothesis.strategies
  from hypothesis import given, assume
  import numpy
  import time
  import os
  from caffe2.proto import caffe2_pb2
  from caffe2.python import core, workspace, muji, dyndep
  import caffe2.python.hypothesis_test_util
label_desc: Use image node_id 1 for calling a global function with example usage: gpu_device(i) and returns: device_option