Searched refs:kernel_module_code (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | metrics.py | 268 def _parse_kernel_fn_code(kernel_module_code): argument
277 mod = PyCodeCache.load(kernel_module_code)
291 def _parse_size_hints(kernel_module_code, kernel_category): argument
295 m = re.search(r"size_hints=(\[[0-9, ]*\]),", kernel_module_code)
300 def _parse_reduction_hint(kernel_category, kernel_module_code): argument
303 m = re.search(r"reduction_hint=ReductionHint\.(\w*),", kernel_module_code)
357 def log_kernel_metadata(kernel_name, kernel_path, kernel_module_code): argument
366 kernel_category = get_kernel_category_by_source_code(kernel_module_code)
367 reduction_hint = _parse_reduction_hint(kernel_category, kernel_module_code)
368 size_hints = _parse_size_hints(kernel_module_code, kernel_category)
[all …]
|