[Metax]Update version information and fix collect_fpn_proposals (#2151)

Author: metax666

.github/workflows/metax_work_private.yaml

@@ -5,8 +5,8 @@ on:
   pull_request:
     types: [opened, synchronize]
     branches: [develop, release/**]
-    schedule:
-      - cron: "0 15 * * *"
+  schedule:
+    - cron: "0 15 * * *"
 permissions: read-all
 
 defaults:

backends/metax_gpu/build_private_CI.sh

@@ -67,7 +67,7 @@ export CUCC_PATH=${MACA_PATH}/tools/cu-bridge
 export PATH=${PATH}:${CUCC_PATH}/tools:${CUCC_PATH}/bin
 export PATH=${MACA_PATH}/bin:${PATH}
 export LD_LIBRARY_PATH=${MACA_PATH}/lib:${MACA_PATH}/mxgpu_llvm/lib:${LD_LIBRARY_PATH}
-export PADDLE_VERSION=3.3.0
+export PADDLE_VERSION="3.3.0.dev$(date +%Y%m%d)"
 
 if [ ! -d build ]; then
     echo "build directory not found, creating..."
@@ -86,6 +86,7 @@ echo "Done!"
 
 cd build/dist/
 ossutil ls oss://opensource-ci/paddle/
-ossutil cat oss://opensource-ci/paddle/test1
-ossutil cp ./paddle_metax_gpu-*.whl oss://opensource-ci/paddle/test1/
+ossutil cat oss://opensource-ci/paddle/
+
+ossutil cp ./paddle_metax_gpu-*.whl oss://opensource-ci/paddle/ -f
 cd -

backends/metax_gpu/compile.sh

@@ -22,7 +22,7 @@ export CUCC_PATH=${MACA_PATH}/tools/cu-bridge
 export PATH=${PATH}:${CUCC_PATH}/tools:${CUCC_PATH}/bin
 export PATH=${MACA_PATH}/bin:${PATH}
 export LD_LIBRARY_PATH=${MACA_PATH}/lib:${MACA_PATH}/mxgpu_llvm/lib:${LD_LIBRARY_PATH}
-
+export PADDLE_VERSION="3.3.0.dev$(date +%Y%m%d)"
 if [ ! -d build ]; then
     echo "build directory not found, creating..."
     mkdir build
@@ -31,7 +31,7 @@ fi
 echo "make_maca"
 cd build
 cmake_maca .. -DCMAKE_BUILD_TYPE=Release -DCMAKE_EXPORT_COMPILE_COMMANDS=ON -DPython3_EXECUTABLE=$(which python3) -DWITH_GPU=ON
-make_maca -j10
+make_maca -j18
 
 
 echo "install whl"

backends/metax_gpu/env.sh

@@ -0,0 +1,20 @@
+# Copyright (c) 2025 PaddlePaddle Authors. All Rights Reserved.
+# 
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+DEFAULT_DIR="/opt/maca"
+export MACA_PATH=${1:-$DEFAULT_DIR}
+export CUDA_PATH=/usr/local/cuda
+export CUCC_PATH=${MACA_PATH}/tools/cu-bridge
+export PATH=${CUDA_PATH}/bin:${MACA_PATH}/ompi/bin:${MACA_PATH}/ucx/bin:${MACA_PATH}/mxgpu_llvm/bin:${MACA_PATH}/bin:${CUCC_PATH}/tools:${CUCC_PATH}/bin:${PATH}
+export LD_LIBRARY_PATH=${MACA_PATH}/lib:${MACA_PATH}/ompi/lib:${MACA_PATH}/mxgpu_llvm/lib:${LD_LIBRARY_PATH}

backends/metax_gpu/kernels/cuda_kernels/collect_fpn_proposals_kernel_register.cu

@@ -12,6 +12,7 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+#include "paddle/phi/core/kernel_registry.h"
 #include "paddle/phi/kernels/gpu/collect_fpn_proposals_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(collect_fpn_proposals,

backends/metax_gpu/setup.py.in

@@ -81,6 +81,11 @@ class BinaryDistribution(Distribution):
     def has_ext_modules(self):
         return True
 
+# maca ai version
+maca_ai_version = os.getenv('MACA_AI_VERSION')
+if not maca_ai_version:
+    maca_ai_version = "0.0.0"
+
 
 def main():
     write_custom_op_api_py()
@@ -89,7 +94,7 @@ def main():
 
     setup(
     name = '@CMAKE_PROJECT_NAME@',
-    version='@PLUGIN_VERSION@',
+    version='@PLUGIN_VERSION@' + "+maca" + maca_ai_version,
     description='Paddle metax_gpu plugin',
     long_description='',
     long_description_content_type="text/markdown",