code: def get_kernel_category_by_source_code(src_code):
      """
      Similar to get_kernel_category but use the source code. Call this API
      if we have not compile the src_code to module yet.
      """
      choices = [
          ch for ch in _kernel_category_choices if f"@{ch}" in src_code
      ]
      if len(choices) == 1:
          return choices[0]
      else:
          return "unknown"
  
file_path: pytorch/torch/_inductor/wrapper_benchmark.py
imports: import dataclasses
  import tempfile
  from collections import defaultdict
  import torch
  from torch.autograd import DeviceType
  from .utils import create_bandwidth_info_str, do_bench, get_num_bytes
label_desc: Use image node_id 1 for calling a global function with example usage: get_kernel_category_by_source_code(src_code) and returns: choices, str