未验证 提交 ce6978c6 编写于 作者: H huangjiyi 提交者: GitHub

update (#52875)

上级 2f499713
...@@ -83,7 +83,7 @@ class KernelSignatureSearcher: ...@@ -83,7 +83,7 @@ class KernelSignatureSearcher:
) )
return pd.merge( return pd.merge(
kernel_func_df, func_signature_df, on='kernel_func', how='left' kernel_func_df, func_signature_df, on='kernel_func', how='left'
)[['kernel_name', 'kernel_signature']] )[['kernel_name', 'kernel_signature']].sort_values(by='kernel_name')
def filter_result(self): def filter_result(self):
for kernel_name in self.filter["kernel_name"]: for kernel_name in self.filter["kernel_name"]:
...@@ -207,14 +207,14 @@ def get_kernel_signatures(): ...@@ -207,14 +207,14 @@ def get_kernel_signatures():
# Because phi/kernels has some independent subdirs, whose kernel names # Because phi/kernels has some independent subdirs, whose kernel names
# (in different namespaces) may conflict with main directory or other # (in different namespaces) may conflict with main directory or other
# subdirs, so we need to search them separately. # subdirs, so we need to search them separately.
indenpendent_subdir = [ independent_subdir = [
'fusion', 'fusion',
'legacy', 'legacy',
'selected_rows', 'selected_rows',
'sparse', 'sparse',
'strings', 'strings',
] ]
for subdir in indenpendent_subdir: for subdir in independent_subdir:
sub_path = osp.join(base_path, subdir) sub_path = osp.join(base_path, subdir)
sub_df = KernelSignatureSearcher.search(sub_path) sub_df = KernelSignatureSearcher.search(sub_path)
kernel_signature_df = pd.concat( kernel_signature_df = pd.concat(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册