diff --git a/paddle/phi/tools/get_kernel_signatures.py b/paddle/phi/tools/get_kernel_signatures.py index cb03c47710a22e874bdde829351b8628314f99aa..1663c446624b93bc90f151bdf7bc5f2c45c9b2f4 100644 --- a/paddle/phi/tools/get_kernel_signatures.py +++ b/paddle/phi/tools/get_kernel_signatures.py @@ -195,8 +195,6 @@ def get_kernel_signatures(): KernelSignatureSearcher.filter["kernel_name"]. """ Paddle_path = osp.abspath(osp.join(osp.dirname(__file__), '../../..')) - if Paddle_path.split('/')[-1] != 'Paddle': - raise ValueError('Paddle path error.') build_path = osp.join(Paddle_path, 'build') os.makedirs(build_path, exist_ok=True) KernelSignatureSearcher.build_path = build_path @@ -209,8 +207,9 @@ def get_kernel_signatures(): # subdirs, so we need to search them separately. independent_subdir = [ 'fusion', - 'legacy', - 'selected_rows', + # Currently, we need filter legacy dir and selected_rows dir. + # 'legacy', + # 'selected_rows', 'sparse', 'strings', ]