code: def num_warps(n):
      if n <= 128:
          return 1
      if n <= 256:
          return 2
      if n <= 512:
          return 4
      if n <= 4096:
          return 8
      return 16
  
file_path: triton/python/triton/ops/blocksparse/softmax.py
imports: import torch
  from ...None import jit
  from ...None import language
  from ...None import next_power_of_2
label_desc: Use image node_id 1 for calling a global function with example usage: num_warps(n) and returns: int, int, int, int, int