Fix cant find nv/targe bug (#73360)

risemeup1 · web-flow · commit d48d3a3c1726 · 2025-06-16T21:04:27.000+08:00
* refine cpp extension

* fix can't find nv/target bug

* fix can't find nv/target bug
diff --git a/paddle/cinn/backends/nvrtc/nvrtc_util.cc b/paddle/cinn/backends/nvrtc/nvrtc_util.cc
@@ -35,6 +35,7 @@ PD_DECLARE_string(cinn_nvcc_cmd_path);
 PD_DECLARE_string(nvidia_package_dir);
 PD_DECLARE_bool(nvrtc_compile_to_cubin);
 PD_DECLARE_bool(cinn_nvrtc_cubin_with_fmad);
+PD_DECLARE_string(cuda_cccl_dir);
 
 namespace cinn {
 namespace backends {
@@ -50,7 +51,8 @@ static std::vector<std::string> GetNvidiaAllIncludePath(
   std::vector<std::string> include_paths;
   const std::string delimiter = "/";
   // Expand this list if necessary.
-  const std::vector<std::string> sub_modules = {"cublas",
+  const std::vector<std::string> sub_modules = {"cuda_cccl",
+                                                "cublas",
                                                 "cudnn",
                                                 "cufft",
                                                 "cusparse",
diff --git a/paddle/common/flags.cc b/paddle/common/flags.cc
@@ -1798,6 +1798,11 @@ PHI_DEFINE_EXPORTED_string(
     "Specify root dir path for nvidia site-package, such as "
     "python3.9/site-packages/nvidia");
 
+PHI_DEFINE_EXPORTED_string(cuda_cccl_dir,  // NOLINT
+                           "",
+                           "Specify root dir path for nv/target, such as "
+                           "python3.9/site-packages/nvidia/cuda_cccl/include/");
+
 PHI_DEFINE_EXPORTED_string(
     cudnn_dir,  // NOLINT
     "",
diff --git a/python/paddle/__init__.py b/python/paddle/__init__.py
@@ -648,6 +648,10 @@
         cupti_dir_lib_path = package_dir + "/.." + "/nvidia/cuda_cupti/lib"
         set_flags({"FLAGS_cupti_dir": cupti_dir_lib_path})
 
+        if is_compiled_with_cinn():
+            cuda_cccl_path = package_dir + "/.." + "/nvidia/cuda_cccl/include/"
+            set_flags({"FLAGS_cuda_cccl_dir": cuda_cccl_path})
+
     elif (
         platform.system() == 'Windows'
         and platform.machine() in ('x86_64', 'AMD64')
diff --git a/python/setup.py.in b/python/setup.py.in
@@ -656,6 +656,22 @@ def get_paddle_extra_install_requirements():
                 "nvidia-cufile-cu12==1.14.0.30; platform_system == 'Linux' and platform_machine == 'x86_64'"
             ),
         }
+        if '@WITH_CINN@' == 'ON':
+            PADDLE_CUDA_INSTALL_REQUIREMENTS["12.3"] += (
+                    " | nvidia-cuda-cccl-cu12==12.3.52;platform_system == 'Linux' and platform_machine == 'x86_64' "
+            )
+            PADDLE_CUDA_INSTALL_REQUIREMENTS["12.4"] += (
+                    " | nvidia-cuda-cccl-cu12==12.4.99;platform_system == 'Linux' and platform_machine == 'x86_64' "
+            )
+            PADDLE_CUDA_INSTALL_REQUIREMENTS["12.6"] += (
+                    " | nvidia-cuda-cccl-cu12==12.6.77;platform_system == 'Linux' and platform_machine == 'x86_64' "
+            )
+            PADDLE_CUDA_INSTALL_REQUIREMENTS["12.8"] += (
+                    " | nvidia-cuda-cccl-cu12==12.8.90;platform_system == 'Linux' and platform_machine == 'x86_64' "
+            )
+            PADDLE_CUDA_INSTALL_REQUIREMENTS["12.9"] += (
+                    " | nvidia-cuda-cccl-cu12==12.9.27;platform_system == 'Linux' and platform_machine == 'x86_64' "
+            )
         elif platform.system() == 'Windows':
             PADDLE_CUDA_INSTALL_REQUIREMENTS = {
                 "11.8": (
diff --git a/setup.py b/setup.py
@@ -1164,6 +1164,23 @@ def get_paddle_extra_install_requirements():
                     "nvidia-cufile-cu12==1.14.0.30; platform_system == 'Linux' and platform_machine == 'x86_64'"
                 ),
             }
+            if env_dict.get("WITH_CINN") == "ON":
+                PADDLE_CUDA_INSTALL_REQUIREMENTS[
+                    "12.3"
+                ] += " | nvidia-cuda-cccl-cu12==12.3.52;platform_system == 'Linux' and platform_machine == 'x86_64' "
+                PADDLE_CUDA_INSTALL_REQUIREMENTS[
+                    "12.4"
+                ] += " | nvidia-cuda-cccl-cu12==12.4.99;platform_system == 'Linux' and platform_machine == 'x86_64' "
+                PADDLE_CUDA_INSTALL_REQUIREMENTS[
+                    "12.6"
+                ] += " | nvidia-cuda-cccl-cu12==12.6.77;platform_system == 'Linux' and platform_machine == 'x86_64' "
+                PADDLE_CUDA_INSTALL_REQUIREMENTS[
+                    "12.8"
+                ] += " | nvidia-cuda-cccl-cu12==12.8.90;platform_system == 'Linux' and platform_machine == 'x86_64' "
+                PADDLE_CUDA_INSTALL_REQUIREMENTS[
+                    "12.9"
+                ] += " | nvidia-cuda-cccl-cu12==12.9.27;platform_system == 'Linux' and platform_machine == 'x86_64' "
+
         elif platform.system() == 'Windows':
             PADDLE_CUDA_INSTALL_REQUIREMENTS = {
                 "11.8": (
