Fix cant find nv/targe bug (#73360)

* refine cpp extension * fix can't find nv/target bug * fix can't find nv/target bug

Author: risemeup1

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",

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
  "",

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')

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": (

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": (