add compat dirs to includes for ci

Author: SigureMo

python/paddle/utils/cpp_extension/extension_utils.py

@@ -835,7 +835,7 @@ def find_paddle_includes(use_cuda=False):
         )
         compat_dir_api_include = os.path.join(
             paddle_include_dir,
-            'paddle/phi/api/include/compat/torch/csrc/api/include/',
+            'paddle/phi/api/include/compat/torch/csrc/api/include',
         )
         include_dirs.extend([compat_dir_root, compat_dir_api_include])
 

test/cpp_extension/utils.py

@@ -12,8 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import os
 import sys
+from pathlib import Path
 from site import getsitepackages
 
 import numpy as np
@@ -28,12 +28,19 @@
 # PaddlePaddle whl. So here we specific `include_dirs` to avoid errors in CI.
 paddle_includes = []
 for site_packages_path in getsitepackages():
-    paddle_includes.append(
-        os.path.join(site_packages_path, 'paddle', 'include')
-    )
-    paddle_includes.append(
-        os.path.join(site_packages_path, 'paddle', 'include', 'third_party')
-    )
+    paddle_include_dir = Path(site_packages_path) / "paddle/include"
+    paddle_includes.append(str(paddle_include_dir))
+    paddle_includes.append(str(paddle_include_dir / 'third_party'))
+    if not IS_WINDOWS:
+        paddle_includes.append(
+            str(paddle_include_dir / 'paddle/phi/api/include/compat')
+        )
+        paddle_includes.append(
+            str(
+                paddle_include_dir
+                / 'paddle/phi/api/include/compat/torch/csrc/api/include'
+            )
+        )
 
 # Test for extra compile args
 extra_cc_args = ['-w', '-g'] if not IS_WINDOWS else ['/w']