Searched refs:proper_kernel_fn_code (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | metrics.py | 284 def _parse_kernel_line_of_code(proper_kernel_fn_code): argument
288 return len(proper_kernel_fn_code.splitlines())
308 def _count_pattern(proper_kernel_fn_code, pattern): argument
309 return proper_kernel_fn_code.count(pattern)
312 def _count_args(proper_kernel_fn_code): argument
313 def_line = proper_kernel_fn_code.splitlines()[0]
330 def _parse_numel(proper_kernel_fn_code, numel_arg_name): argument
331 m = re.search(f"{numel_arg_name} = ([\\d]+)", proper_kernel_fn_code)
371 proper_kernel_fn_code = _parse_proper_kernel_fn_code(kernel_fn_code)
374 kernel_line_of_code = _parse_kernel_line_of_code(proper_kernel_fn_code)
[all …]
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_metrics.py | 60 proper_kernel_fn_code = metrics._parse_proper_kernel_fn_code(example_kernel)
61 assert proper_kernel_fn_code.startswith("def ")
64 proper_kernel_fn_code = metrics._parse_proper_kernel_fn_code(example_kernel)
65 self.assertEqual(6, metrics._count_args(proper_kernel_fn_code))
68 proper_kernel_fn_code = metrics._parse_proper_kernel_fn_code(example_kernel)
69 self.assertEqual(2, metrics._count_pattern(proper_kernel_fn_code, "tl.load"))
70 self.assertEqual(1, metrics._count_pattern(proper_kernel_fn_code, "tl.store"))
71 self.assertEqual(1, metrics._count_pattern(proper_kernel_fn_code, "for "))
|