code: def set_device(device):
      global DEVICE
      DEVICE = device
  
file_path: text-generation-inference/server/text_generation_server/utils/gptq/exllama.py
imports: import torch
  from exllama_kernels import make_q4, q4_matmul, prepare_buffers, set_tuning_params
label_desc: Use image node_id 3 for calling a global function with example usage: set_device(device) without return types