apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-12-24 13:28:13 +08:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

【Hackathon 9th No.109】[CppExtension] Support build Custom OP in setuptools 80+ (#4977)

Browse Source 
* 添加对现代Python打包方法的兼容性支持

* [CppExtension] 优化构建脚本逻辑并更新.gitignore
This commit is contained in:
megemini
2025-11-17 11:46:27 +08:00
committed by GitHub
parent 02c83d65db
commit c35e540c18
2 changed files with 29 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@@ -136,6 +136,7 @@ kernel_meta*
fastdeploy_ops.py
version.txt
EGG-INFO/
**/fastdeploy_ops/__init__.py
# fp8 generated codes
autogen/

28
build.sh
View File

@@ -79,6 +79,34 @@ function copy_ops(){
PROCESSOR_VERSION=`${python} -c "import platform; print(platform.processor())"`
WHEEL_NAME="fastdeploy_ops-${OPS_VERSION}-${PY_VERSION}-${SYSTEM_VERSION}-${PROCESSOR_VERSION}.egg"
WHEEL_CPU_NAME="fastdeploy_cpu_ops-${OPS_VERSION}-${PY_VERSION}-${SYSTEM_VERSION}-${PROCESSOR_VERSION}.egg"
# Add compatibility for modern python packaging methods
WHEEL_MODERN_NAME="fastdeploy_ops"
WHEEL_MODERN_CPU_NAME="fastdeploy_cpu_ops"
# Handle GPU ops directories (WHEEL_MODERN_NAME and WHEEL_NAME)
if [ -d "./${OPS_TMP_DIR}/${WHEEL_MODERN_NAME}" ]; then
echo -e "${GREEN}[Info]${NONE} Ready to copy ops from modern directory ${WHEEL_MODERN_NAME} to target directory"
# Set WHEEL_NAME to empty string to ignore the directory path
WHEEL_NAME=""
else
# If modern directory doesn't exist, check for deprecated directory
if [ -d "./${OPS_TMP_DIR}/${WHEEL_NAME}" ]; then
echo -e "${YELLOW}[Warning]${NONE} ${WHEEL_NAME} directory exists. This is a deprecated packaging and distribution method."
fi
fi
# Handle CPU ops directories (WHEEL_MODERN_CPU_NAME and WHEEL_CPU_NAME)
if [ -d "./${OPS_TMP_DIR}/${WHEEL_MODERN_CPU_NAME}" ]; then
echo -e "${GREEN}[Info]${NONE} Ready to copy ops from modern directory ${WHEEL_MODERN_CPU_NAME} to target directory"
# Set WHEEL_CPU_NAME to empty string to ignore the directory path
WHEEL_CPU_NAME=""
else
# If modern directory doesn't exist, check for deprecated directory
if [ -d "./${OPS_TMP_DIR}/${WHEEL_CPU_NAME}" ]; then
echo -e "${YELLOW}[Warning]${NONE} ${WHEEL_CPU_NAME} directory exists. This is a deprecated packaging and distribution method."
fi
fi
is_rocm=`$python -c "import paddle; print(paddle.is_compiled_with_rocm())"`
if [ "$is_rocm" = "True" ]; then
DEVICE_TYPE="rocm"