fix(ascend): use image-provided ATB runtime

Author: AlexMa616

.github/configs/ascend.yml

@@ -33,7 +33,6 @@ container_volumes:
   - /public-ks3/cicd/baai_tokenizers:/home/gitlab-runner/tokenizers
   - /usr/local/Ascend/driver:/usr/local/Ascend/driver:ro
   - /usr/local/Ascend/add-ons:/usr/local/Ascend/add-ons:ro
-  - /usr/local/Ascend/nnal:/usr/local/Ascend/nnal:ro
   - /usr/local/sbin/npu-smi:/usr/local/sbin/npu-smi:ro
 
 # Image build contract consumed by the generic build workflow.

.github/workflows/functional_tests_inference.yml

@@ -230,6 +230,10 @@ jobs:
               ;;
             pip)
               echo "ℹ️  Running tests with pip/system Python"
+              PLATFORM_ENV_SCRIPT="./tools/install/$PLATFORM/env.sh"
+              if [ -f "$PLATFORM_ENV_SCRIPT" ]; then
+                source "$PLATFORM_ENV_SCRIPT"
+              fi
               ;;
           esac
 

.github/workflows/functional_tests_serve.yml

@@ -233,6 +233,10 @@ jobs:
               ;;
             pip)
               echo "Running tests with pip/system Python"
+              PLATFORM_ENV_SCRIPT="./tools/install/$PLATFORM/env.sh"
+              if [ -f "$PLATFORM_ENV_SCRIPT" ]; then
+                source "$PLATFORM_ENV_SCRIPT"
+              fi
               ;;
           esac
 

tools/install/ascend/activate_runtime.sh

@@ -35,7 +35,10 @@ export ASCEND_OPP_PATH="$toolkit_home/opp"
 # inference image's Python/CANN libraries (or vice versa).
 export PATH="$python_home/bin:$toolkit_home/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin"
 nnal_library_paths=""
-for library_path in "$ascend_home"/nnal/atb/*/lib "$ascend_home"/nnal/asdsip/*/lib; do
+for library_path in \
+    "$ascend_home"/nnal/atb/*/atb/cxx_abi_*/lib \
+    "$ascend_home"/nnal/atb/*/lib \
+    "$ascend_home"/nnal/asdsip/*/lib; do
     [ -d "$library_path" ] || continue
     nnal_library_paths="$nnal_library_paths:$library_path"
 done

tools/install/ascend/env.sh

@@ -66,11 +66,21 @@
 # -----------------------------------------------------------------------------
 export FLAGSCALE_HOME FLAGSCALE_CONDA FLAGSCALE_DEPS FLAGSCALE_DOWNLOADS
 export UV_PROJECT_ENVIRONMENT MPI_HOME ASCEND_HOME ASCEND_TOOLKIT_HOME
+export ASCEND_HOME_PATH="$ASCEND_TOOLKIT_HOME"
+export ASCEND_OPP_PATH="$ASCEND_TOOLKIT_HOME/opp"
 export UV_HTTP_TIMEOUT UV_INDEX_STRATEGY UV_LINK_MODE
 export VIRTUAL_ENV="$UV_PROJECT_ENVIRONMENT"
 
 # -----------------------------------------------------------------------------
 # PATH Configuration
 # -----------------------------------------------------------------------------
 export PATH="$UV_PROJECT_ENVIRONMENT/bin:$FLAGSCALE_CONDA/bin:$HOME/.local/bin:$MPI_HOME/bin:$ASCEND_TOOLKIT_HOME/bin:$PATH"
-export LD_LIBRARY_PATH="$ASCEND_TOOLKIT_HOME/lib64:$ASCEND_HOME/driver/lib64:$MPI_HOME/lib64:$MPI_HOME/lib:/usr/local/lib:$LD_LIBRARY_PATH"
+nnal_library_paths=""
+for library_path in \
+    "$ASCEND_HOME"/nnal/atb/*/atb/cxx_abi_*/lib \
+    "$ASCEND_HOME"/nnal/atb/*/lib \
+    "$ASCEND_HOME"/nnal/asdsip/*/lib; do
+    [ -d "$library_path" ] || continue
+    nnal_library_paths="$nnal_library_paths:$library_path"
+done
+export LD_LIBRARY_PATH="$ASCEND_TOOLKIT_HOME/lib64$nnal_library_paths:$ASCEND_HOME/driver/lib64:$MPI_HOME/lib64:$MPI_HOME/lib:/usr/local/lib:${LD_LIBRARY_PATH:-}"

tools/install/ascend/validate_image_build.sh

@@ -22,7 +22,6 @@ docker_args=(
     --device /dev/hisi_hdc
     --volume /usr/local/Ascend/driver:/usr/local/Ascend/driver:ro
     --volume /usr/local/Ascend/add-ons:/usr/local/Ascend/add-ons:ro
-    --volume /usr/local/Ascend/nnal:/usr/local/Ascend/nnal:ro
     --volume /usr/local/sbin/npu-smi:/usr/local/sbin/npu-smi:ro
     --privileged
 )