diff --git a/.github/workflows/test_cc.yml b/.github/workflows/test_cc.yml index 9b803131e2..03192cea79 100644 --- a/.github/workflows/test_cc.yml +++ b/.github/workflows/test_cc.yml @@ -41,7 +41,10 @@ jobs: python-version: "3.11" cache: "pip" - uses: lukka/get-cmake@latest - - run: python -m pip install uv + - name: Install uv + uses: astral-sh/setup-uv@v7 + with: + enable-cache: true - name: Install Python dependencies run: | source/install/uv_with_retry.sh pip install --system --group pin_tensorflow_cpu --group pin_pytorch_cpu --group pin_jax_cpu --torch-backend cpu @@ -63,6 +66,7 @@ jobs: TF_INTRA_OP_PARALLELISM_THREADS: 1 TF_INTER_OP_PARALLELISM_THREADS: 1 CMAKE_GENERATOR: Ninja + CTEST_PARALLEL_LEVEL: 3 CXXFLAGS: ${{ matrix.check_memleak && '-fsanitize=leak' || '' }} LSAN_OPTIONS: suppressions=${{ github.workspace }}/.github/workflows/suppr.txt ENABLE_TENSORFLOW: ${{ matrix.enable_tensorflow && 'TRUE' || 'FALSE' }} diff --git a/.github/workflows/test_cuda.yml b/.github/workflows/test_cuda.yml index 4be035ba21..041ed7ccf8 100644 --- a/.github/workflows/test_cuda.yml +++ b/.github/workflows/test_cuda.yml @@ -56,7 +56,8 @@ jobs: DP_ENABLE_NATIVE_OPTIMIZATION: 1 DP_ENABLE_PYTORCH: 1 - run: dp --version - - run: python -m pytest source/tests + - name: Test Python on CUDA + run: python -m pytest source/tests --ignore=source/tests/pt_expt env: NUM_WORKERS: 0 CUDA_VISIBLE_DEVICES: 0 diff --git a/.github/workflows/test_python.yml b/.github/workflows/test_python.yml index 3ace62c9e3..d45a8e8686 100644 --- a/.github/workflows/test_python.yml +++ b/.github/workflows/test_python.yml @@ -26,7 +26,10 @@ jobs: - uses: actions/setup-python@v6 with: python-version: ${{ matrix.python }} - - run: python -m pip install -U uv + - name: Install uv + uses: astral-sh/setup-uv@v7 + with: + enable-cache: true - run: | source/install/uv_with_retry.sh pip install --system openmpi --group pin_tensorflow_cpu --group pin_pytorch_cpu --torch-backend cpu export TENSORFLOW_ROOT=$(python -c 'import importlib.util,pathlib;print(pathlib.Path(importlib.util.find_spec("tensorflow").origin).parent)') diff --git a/source/install/test_cc_local.sh b/source/install/test_cc_local.sh index 5ddbf0eecc..70f1cbda12 100755 --- a/source/install/test_cc_local.sh +++ b/source/install/test_cc_local.sh @@ -102,4 +102,4 @@ if [ "${ENABLE_PADDLE:-TRUE}" == "TRUE" ]; then PADDLE_INFERENCE_DIR=${BUILD_TMP_DIR}/paddle_inference_install_dir export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:${PADDLE_INFERENCE_DIR}/third_party/install/onednn/lib:${PADDLE_INFERENCE_DIR}/third_party/install/mklml/lib fi -ctest --output-on-failure +ctest --output-on-failure --parallel "${CTEST_PARALLEL_LEVEL:-1}"