code: def get_depth(args):
      return triton.next_power_of_2(args["K"])
  
imports: import triton
  import triton.language
label_desc: Use image node_id 1 for calling a global function with example usage: get_depth(args) and returns: triton