Home
last modified time | relevance | path

Searched refs:kernel_fn_code (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dmetrics.py322 def _parse_proper_kernel_fn_code(kernel_fn_code): argument
326 start_pos = kernel_fn_code.index("def ")
327 return kernel_fn_code[start_pos:]
338 def _parse_kernel_args_num_gb(kernel_fn_code, kernel_category): argument
343 m = re.search(r".kernel_num_gb.:\s*([0-9.]+)", kernel_fn_code)
369 kernel_fn_code = _parse_kernel_fn_code(kernel_module_code)
371 proper_kernel_fn_code = _parse_proper_kernel_fn_code(kernel_fn_code)
393 kernel_fn_code, kernel_category