Searched refs:get_kernel_category_by_source_code (Results 1 – 4 of 4) sorted by relevance
6 from torch._inductor.wrapper_benchmark import get_kernel_category_by_source_code74 kernel_category = get_kernel_category_by_source_code(example_kernel)
364 from .wrapper_benchmark import get_kernel_category_by_source_code366 kernel_category = get_kernel_category_by_source_code(kernel_module_code)
23 def get_kernel_category_by_source_code(src_code): function
54 from ..wrapper_benchmark import get_kernel_category_by_source_code3021 kernel_category = get_kernel_category_by_source_code(src_code)[:3]