code: def num_warps(n):
      if n < 512:
          return 4
      if n < 2048:
          return 8
      return 16
  
file_path: DeepSpeed/deepspeed/ops/sparse_attention/softmax.py
imports: import torch
  import triton
  import triton.language
label_desc: Use image node_id 2 for calling a global function with example usage: num_warps(n) and returns: int, int, int