code: def setUp(self):
      super().setUp()
      gpus = config.list_physical_devices("GPU")
      config.set_logical_device_configuration(
          gpus[0],
          [
              context.LogicalDeviceConfiguration(memory_limit=1024),
              context.LogicalDeviceConfiguration(memory_limit=1024),
              context.LogicalDeviceConfiguration(memory_limit=1024),
          ],
      )
  
file_path: tensorflow/tensorflow/python/compiler/xla/pjrt_compile_virtual_device_test.py
imports: from tensorflow.python.eager import context
  from tensorflow.python.eager import def_function
  from tensorflow.python.eager import test
  from tensorflow.python.framework import config
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import ops
  from tensorflow.python.ops import variables
label_desc: Use image node_id 1 for calling the PjrtCompileVirtualDeviceTest obj's underlying member method code with example usage: obj.setUp() without return types