def _identify_trace_mode()

in sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py [0:0]

• 11 lines of code
• 1 McCabe index (conditional complexity)

def _identify_trace_mode():
  global _TRACE_MODE
  _TRACE_MODE = os.environ.get('TORCH_DISTRIBUTED_TRACING', 'CROSSNODE')
  _TRACE_MODE = _TRACE_MODE.lower()

  global _SHOULD_PRINT
  _SHOULD_PRINT = os.environ.get('TORCH_DISTRIBUTED_TRACING_PRINT', 'False')
  _SHOULD_PRINT = _SHOULD_PRINT.lower() in ['true', '1', 't', 'y', 'yes']

  global _GPU_SERIAL
  _GPU_SERIAL = os.environ.get("GPU_SERIAL", "unknown")
  global _VM_ID
  _VM_ID = os.environ.get("VM_ID", "unknown")