From 3aed8d257de46c7ec8acfa54e8259a988cb77624 Mon Sep 17 00:00:00 2001 From: Lucas Date: Fri, 5 Dec 2025 17:34:17 +0800 Subject: [PATCH] [XPU] redirect xvllm/xtdk/xhpc downloading log (#5388) --- custom_ops/xpu_ops/download_dependencies.sh | 4 ++-- custom_ops/xpu_ops/setup_ops.py | 10 ++++------ 2 files changed, 6 insertions(+), 8 deletions(-) diff --git a/custom_ops/xpu_ops/download_dependencies.sh b/custom_ops/xpu_ops/download_dependencies.sh index 7e5ff7586..ad6d4d2de 100644 --- a/custom_ops/xpu_ops/download_dependencies.sh +++ b/custom_ops/xpu_ops/download_dependencies.sh @@ -26,14 +26,14 @@ fi rm -rf output* xvllm* xtdk-llvm* output.tar.gz xtdk-llvm*tar.gz # Download and install xvllm - if ! wget "https://klx-sdk-release-public.su.bcebos.com/xinfer/daily/eb/${version_xvllm}/output.tar.gz"; then + if ! wget -q "https://klx-sdk-release-public.su.bcebos.com/xinfer/daily/eb/${version_xvllm}/output.tar.gz"; then echo "Error downloading xvllm" exit 2 fi tar -zxf output.tar.gz && mv output xvllm && rm output.tar.gz # Download and install xtdk - if ! wget "https://klx-sdk-release-public.su.bcebos.com/xtdk_15fusion/dev/${version_xtdk}/xtdk-llvm15-ubuntu2004_x86_64.tar.gz"; then + if ! wget -q "https://klx-sdk-release-public.su.bcebos.com/xtdk_15fusion/dev/${version_xtdk}/xtdk-llvm15-ubuntu2004_x86_64.tar.gz"; then echo "Error downloading xtdk" exit 3 fi diff --git a/custom_ops/xpu_ops/setup_ops.py b/custom_ops/xpu_ops/setup_ops.py index d29e6290f..fa8d95b1d 100755 --- a/custom_ops/xpu_ops/setup_ops.py +++ b/custom_ops/xpu_ops/setup_ops.py @@ -48,6 +48,8 @@ def download_and_extract(url, destination_directory): print(f"Downloading: {url}") subprocess.run( ["wget", "-O", file_path, url], + stdout=subprocess.DEVNULL, + stderr=subprocess.DEVNULL, check=True, ) print(f"Downloaded to: {file_path}") @@ -152,12 +154,8 @@ def xpu_setup_ops(): XDNN_PATH = os.getenv("XDNN_PATH") if XDNN_PATH is None: - version_cmd = 'python -c "import paddle; print(paddle.version.xpu_xhpc())"' - try: - XHPC_VERSION = subprocess.check_output(version_cmd, shell=True).strip().decode() - print(f"Fetched XHPC_VERSION from paddle: {XHPC_VERSION}") - except Exception as e: - raise Exception(f"PaddlePaddle-xpu not installed, please install it first. {e}") + XHPC_VERSION = paddle.version.xpu_xhpc() + print(f"Fetched XHPC_VERSION from paddle: {XHPC_VERSION}") XHPC_URL = f"https://klx-sdk-release-public.su.bcebos.com/xhpc/{XHPC_VERSION}/xhpc-ubuntu2004_x86_64.tar.gz" THIRD_PARTY_PATH = os.path.join(current_file.parent, "third_party")