mirror of
https://github.com/Tencent/ncnn.git
synced 2026-10-17 04:41:47 +08:00
nihui
392 lines
18 KiB
YAML
392 lines
18 KiB
YAML
name: pnnx
|
|
on:
|
|
push:
|
|
branches: [master]
|
|
paths:
|
|
- '.github/workflows/pnnx.yml'
|
|
- 'src/layer/*'
|
|
- 'tools/pnnx/**'
|
|
- '!tools/pnnx/README.md'
|
|
pull_request:
|
|
branches: [master]
|
|
paths:
|
|
- '.github/workflows/pnnx.yml'
|
|
- 'src/layer/*'
|
|
- 'tools/pnnx/**'
|
|
- '!tools/pnnx/README.md'
|
|
concurrency:
|
|
group: pnnx-${{ github.ref }}
|
|
cancel-in-progress: true
|
|
permissions:
|
|
contents: read
|
|
|
|
env:
|
|
LIBTORCH_VERSION: 2.14.0
|
|
TORCHVISION_VERSION: 0.29.0
|
|
PROTOBUF_VERSION: 33.6
|
|
ONNXRUNTIME_VERSION: 1.30.0
|
|
CACHE_DATE: 20260916
|
|
SEGMENT_DOWNLOAD_TIMEOUT_MINS: 15
|
|
|
|
jobs:
|
|
quick-test:
|
|
runs-on: ${{ matrix.os }}
|
|
strategy:
|
|
fail-fast: false
|
|
matrix:
|
|
os: [ubuntu-latest, macos-latest, windows-latest]
|
|
|
|
env:
|
|
PYTHONUSERBASE: ${{ github.workspace }}/torch
|
|
UseMultiToolTask: true
|
|
steps:
|
|
- uses: actions/checkout@v7
|
|
|
|
- uses: actions/setup-python@v7
|
|
with:
|
|
python-version: 3.12
|
|
|
|
- name: setup-pytorch
|
|
run: |
|
|
python3 -m pip config set global.break-system-packages true
|
|
pip3 install --user torch --index-url https://download.pytorch.org/whl/cpu
|
|
pip3 install --user numpy packaging
|
|
|
|
- name: build-pnnx
|
|
run: |
|
|
cd tools/pnnx
|
|
mkdir build && cd build
|
|
cmake -DCMAKE_BUILD_TYPE=Release ..
|
|
cmake --build . --config Release -j 4
|
|
|
|
- name: quick-test
|
|
if: matrix.os != 'windows-latest'
|
|
run: |
|
|
cd tools/pnnx
|
|
cd build && ctest -C Release --output-on-failure -R test_nn_Conv
|
|
|
|
build:
|
|
runs-on: [self-hosted, linux, ubuntu25]
|
|
|
|
steps:
|
|
- uses: actions/checkout@v7
|
|
|
|
- name: local-cache-libtorch
|
|
id: local-cache-libtorch
|
|
uses: maxnowack/local-cache@v3
|
|
with:
|
|
path: libtorch-${{ env.LIBTORCH_VERSION }}-install
|
|
key: libtorch-${{ env.LIBTORCH_VERSION }}-linux-install-${{ env.CACHE_DATE }}
|
|
|
|
- name: local-cache-torchvision
|
|
id: local-cache-torchvision
|
|
uses: maxnowack/local-cache@v3
|
|
with:
|
|
path: torchvision-${{ env.TORCHVISION_VERSION }}-install
|
|
key: torchvision-${{ env.TORCHVISION_VERSION }}-linux-install-${{ env.CACHE_DATE }}
|
|
|
|
- name: local-cache-onnxruntime
|
|
id: local-cache-onnxruntime
|
|
uses: maxnowack/local-cache@v3
|
|
with:
|
|
path: onnxruntime-${{ env.ONNXRUNTIME_VERSION }}-install
|
|
key: onnxruntime-${{ env.ONNXRUNTIME_VERSION }}-linux-install-${{ env.CACHE_DATE }}
|
|
|
|
- name: cache-libtorch
|
|
id: cache-libtorch
|
|
uses: actions/cache@v6
|
|
with:
|
|
path: libtorch-${{ env.LIBTORCH_VERSION }}-install
|
|
key: libtorch-${{ env.LIBTORCH_VERSION }}-linux-install-${{ env.CACHE_DATE }}
|
|
|
|
- name: cache-torchvision
|
|
id: cache-torchvision
|
|
uses: actions/cache@v6
|
|
with:
|
|
path: torchvision-${{ env.TORCHVISION_VERSION }}-install
|
|
key: torchvision-${{ env.TORCHVISION_VERSION }}-linux-install-${{ env.CACHE_DATE }}
|
|
|
|
- name: cache-onnxruntime
|
|
id: cache-onnxruntime
|
|
uses: actions/cache@v6
|
|
with:
|
|
path: onnxruntime-${{ env.ONNXRUNTIME_VERSION }}-install
|
|
key: onnxruntime-${{ env.ONNXRUNTIME_VERSION }}-linux-install-${{ env.CACHE_DATE }}
|
|
|
|
- name: pnnx-patches
|
|
if: (steps.local-cache-libtorch.outputs.cache-hit != 'true' && steps.cache-libtorch.outputs.cache-hit != 'true') || (steps.local-cache-torchvision.outputs.cache-hit != 'true' && steps.cache-torchvision.outputs.cache-hit != 'true') || (steps.local-cache-onnxruntime.outputs.cache-hit != 'true' && steps.cache-onnxruntime.outputs.cache-hit != 'true')
|
|
uses: actions/checkout@v7
|
|
with:
|
|
repository: pnnx/pnnx
|
|
path: pnnx-patches
|
|
|
|
- name: libtorch
|
|
if: steps.local-cache-libtorch.outputs.cache-hit != 'true' && steps.cache-libtorch.outputs.cache-hit != 'true'
|
|
run: |
|
|
wget -q https://github.com/pytorch/pytorch/releases/download/v${{ env.LIBTORCH_VERSION }}/pytorch-v${{ env.LIBTORCH_VERSION }}.tar.gz
|
|
tar -xf pytorch-v${{ env.LIBTORCH_VERSION }}.tar.gz
|
|
cd pytorch-v${{ env.LIBTORCH_VERSION }}
|
|
pip3 install -r requirements.txt --break-system-packages
|
|
patch -p1 -i $GITHUB_WORKSPACE/pnnx-patches/pytorch-v${{ env.LIBTORCH_VERSION }}-fix-mobile-build.patch
|
|
patch -p1 -i $GITHUB_WORKSPACE/pnnx-patches/pytorch-v${{ env.LIBTORCH_VERSION }}-fix-stable-abi.patch
|
|
patch -p1 -i $GITHUB_WORKSPACE/pnnx-patches/pytorch-v${{ env.LIBTORCH_VERSION }}-no-link-system-lib.patch
|
|
patch -p1 -i $GITHUB_WORKSPACE/pnnx-patches/pytorch-v${{ env.LIBTORCH_VERSION }}-set-python-executable.patch
|
|
patch -p1 -i $GITHUB_WORKSPACE/pnnx-patches/pytorch-v${{ env.LIBTORCH_VERSION }}-fix-eigen-build.patch
|
|
patch -p1 -i $GITHUB_WORKSPACE/pnnx-patches/pytorch-v${{ env.LIBTORCH_VERSION }}-fix-link-local-sleef.patch
|
|
patch -p1 -i $GITHUB_WORKSPACE/pnnx-patches/pytorch-v${{ env.LIBTORCH_VERSION }}-revert-nativert-api.patch
|
|
mkdir -p build && cd build
|
|
cmake -DCMAKE_INSTALL_PREFIX=$GITHUB_WORKSPACE/libtorch-${{ env.LIBTORCH_VERSION }}-install \
|
|
-DCMAKE_BUILD_TYPE=MinSizeRel \
|
|
-DBUILD_SHARED_LIBS=OFF \
|
|
-DCMAKE_POLICY_VERSION_MINIMUM=3.5 \
|
|
-DBUILD_CUSTOM_PROTOBUF=OFF \
|
|
-DBUILD_LITE_INTERPRETER=OFF \
|
|
-DBUILD_PYTHON=OFF \
|
|
-DINTERN_BUILD_MOBILE=ON \
|
|
-DINTERN_DISABLE_AUTOGRAD=ON \
|
|
-DINTERN_DISABLE_ONNX=ON \
|
|
-DUSE_CUDA=OFF \
|
|
-DUSE_DISTRIBUTED=OFF \
|
|
-DUSE_ITT=OFF \
|
|
-DUSE_KINETO=OFF \
|
|
-DUSE_LITE_INTERPRETER_PROFILER=OFF \
|
|
-DUSE_MKLDNN=OFF \
|
|
-DUSE_MPS=OFF \
|
|
-DUSE_NUMPY=OFF \
|
|
-DUSE_OPENMP=OFF \
|
|
-DUSE_SOURCE_DEBUG_ON_MOBILE=OFF \
|
|
-DUSE_XNNPACK=OFF \
|
|
-DBUILD_TEST=OFF \
|
|
-DATEN_NO_TEST=ON \
|
|
..
|
|
cmake --build . -j 8
|
|
cmake --build . -j 8 --target install/strip
|
|
|
|
- name: torchvision
|
|
if: steps.local-cache-torchvision.outputs.cache-hit != 'true' && steps.cache-torchvision.outputs.cache-hit != 'true'
|
|
run: |
|
|
wget -q https://github.com/pytorch/vision/archive/v${{ env.TORCHVISION_VERSION }}.zip -O vision-${{ env.TORCHVISION_VERSION }}.zip
|
|
unzip -q vision-${{ env.TORCHVISION_VERSION }}.zip
|
|
cd vision-${{ env.TORCHVISION_VERSION }}
|
|
patch -p1 -i $GITHUB_WORKSPACE/pnnx-patches/vision-${{ env.TORCHVISION_VERSION }}-ops-only.patch
|
|
patch -p1 -i $GITHUB_WORKSPACE/pnnx-patches/vision-${{ env.TORCHVISION_VERSION }}-no-cuda-version.patch
|
|
mkdir -p build && cd build
|
|
cmake -DCMAKE_INSTALL_PREFIX=$GITHUB_WORKSPACE/torchvision-${{ env.TORCHVISION_VERSION }}-install \
|
|
-DTorch_DIR=$GITHUB_WORKSPACE/libtorch-${{ env.LIBTORCH_VERSION }}-install/share/cmake/Torch \
|
|
-DCMAKE_BUILD_TYPE=MinSizeRel \
|
|
-DWITH_PNG=OFF \
|
|
-DWITH_JPEG=OFF ..
|
|
cmake --build . -j 8
|
|
cmake --build . -j 8 --target install/strip
|
|
|
|
- name: onnxruntime
|
|
if: steps.local-cache-onnxruntime.outputs.cache-hit != 'true' && steps.cache-onnxruntime.outputs.cache-hit != 'true'
|
|
run: |
|
|
wget -q https://github.com/protocolbuffers/protobuf/archive/v${{ env.PROTOBUF_VERSION }}.zip -O protobuf-${{ env.PROTOBUF_VERSION }}.zip
|
|
unzip -q protobuf-${{ env.PROTOBUF_VERSION }}.zip
|
|
cd protobuf-${{ env.PROTOBUF_VERSION }}
|
|
patch -p1 -i $GITHUB_WORKSPACE/pnnx-patches/protobuf-${{ env.PROTOBUF_VERSION }}-align-abseil.patch
|
|
mkdir -p build2 && cd build2
|
|
cmake -DCMAKE_INSTALL_PREFIX=$GITHUB_WORKSPACE/onnxruntime-${{ env.ONNXRUNTIME_VERSION }}-install \
|
|
-Dprotobuf_BUILD_TESTS=OFF \
|
|
-DCMAKE_BUILD_TYPE=MinSizeRel \
|
|
-DCMAKE_POSITION_INDEPENDENT_CODE=ON \
|
|
-DCMAKE_INSTALL_LIBDIR=lib \
|
|
-DCMAKE_CXX_STANDARD=20 \
|
|
-Dprotobuf_BUILD_SHARED_LIBS=OFF \
|
|
-Dprotobuf_WITH_ZLIB=OFF ..
|
|
cmake --build . -j 8
|
|
cmake --build . -j 8 --target install/strip
|
|
|
|
cd ../../
|
|
wget -q https://github.com/microsoft/onnxruntime/archive/v${{ env.ONNXRUNTIME_VERSION }}.zip -O onnxruntime-${{ env.ONNXRUNTIME_VERSION }}.zip
|
|
unzip -q onnxruntime-${{ env.ONNXRUNTIME_VERSION }}.zip
|
|
cd onnxruntime-${{ env.ONNXRUNTIME_VERSION }}
|
|
patch -p1 -i $GITHUB_WORKSPACE/pnnx-patches/onnxruntime-${{ env.ONNXRUNTIME_VERSION }}-less-mlas-features.patch
|
|
patch -p1 -i $GITHUB_WORKSPACE/pnnx-patches/onnxruntime-${{ env.ONNXRUNTIME_VERSION }}-monolithic-static-library.patch
|
|
patch -p1 -i $GITHUB_WORKSPACE/pnnx-patches/onnxruntime-${{ env.ONNXRUNTIME_VERSION }}-use-clog.patch
|
|
mkdir -p build2 && cd build2
|
|
cmake -DCMAKE_INSTALL_PREFIX=$GITHUB_WORKSPACE/onnxruntime-${{ env.ONNXRUNTIME_VERSION }}-install \
|
|
-DCMAKE_BUILD_TYPE=MinSizeRel \
|
|
-Donnxruntime_USE_FULL_PROTOBUF=ON \
|
|
-Donnxruntime_BUILD_SHARED_LIB=ON \
|
|
-Donnxruntime_BUILD_UNIT_TESTS=OFF \
|
|
-Donnxruntime_ENABLE_CPUINFO=OFF \
|
|
-Donnxruntime_DISABLE_CONTRIB_OPS=ON \
|
|
-Donnxruntime_DISABLE_ML_OPS=ON \
|
|
-Donnxruntime_DISABLE_SPARSE_TENSORS=ON \
|
|
-DCMAKE_POSITION_INDEPENDENT_CODE=ON \
|
|
-DCMAKE_INSTALL_LIBDIR=lib \
|
|
-DCMAKE_PREFIX_PATH="$GITHUB_WORKSPACE/onnxruntime-${{ env.ONNXRUNTIME_VERSION }}-install" \
|
|
-DCMAKE_FIND_PACKAGE_PREFER_CONFIG=ON \
|
|
-DONNX_CUSTOM_PROTOC_EXECUTABLE="$GITHUB_WORKSPACE/onnxruntime-${{ env.ONNXRUNTIME_VERSION }}-install/bin/protoc" \
|
|
--compile-no-warning-as-error ../cmake
|
|
cmake --build . -j 8
|
|
cmake --build . -j 8 --target install/strip
|
|
|
|
- name: pnnx
|
|
run: |
|
|
cd tools/pnnx
|
|
mkdir build && cd build
|
|
cmake -DCMAKE_BUILD_TYPE=MinSizeRel \
|
|
-DTorch_INSTALL_DIR=$GITHUB_WORKSPACE/libtorch-${{ env.LIBTORCH_VERSION }}-install \
|
|
-DTorchVision_INSTALL_DIR=$GITHUB_WORKSPACE/torchvision-${{ env.TORCHVISION_VERSION }}-install \
|
|
-Donnxruntime_INSTALL_DIR=$GITHUB_WORKSPACE/onnxruntime-${{ env.ONNXRUNTIME_VERSION }}-install \
|
|
-DCMAKE_PREFIX_PATH=$GITHUB_WORKSPACE/onnxruntime-${{ env.ONNXRUNTIME_VERSION }}-install \
|
|
-Dprotobuf_DIR=$GITHUB_WORKSPACE/onnxruntime-${{ env.ONNXRUNTIME_VERSION }}-install/lib/cmake/protobuf ..
|
|
cmake --build . -j 8
|
|
strip src/pnnx
|
|
|
|
- name: upload-pnnx
|
|
uses: actions/upload-artifact@v7
|
|
with:
|
|
name: pnnx
|
|
path: tools/pnnx/build/src/pnnx
|
|
compression-level: 9
|
|
|
|
test:
|
|
needs: [build]
|
|
runs-on: [self-hosted, linux, ubuntu25]
|
|
strategy:
|
|
fail-fast: false
|
|
matrix:
|
|
include:
|
|
- { python: '3.8', numpy: '1.24.4', opencv: '4.5.*', torch: '1.8.1', torchvision: '0.9.1', torchaudio: '0.8.1', transformers: '4.52.1' }
|
|
- { python: '3.8', numpy: '1.24.4', opencv: '4.5.*', torch: '1.9.1', torchvision: '0.10.1', torchaudio: '0.9.1', transformers: '4.52.1' }
|
|
- { python: '3.8', numpy: '1.24.4', opencv: '4.6.*', torch: '1.10.0', torchvision: '0.11.1', torchaudio: '0.10.0+cpu', transformers: '4.52.1' }
|
|
- { python: '3.9', numpy: '1.26.4', opencv: '4.6.*', torch: '1.11.0', torchvision: '0.12.0', torchaudio: '0.11.0+cpu', transformers: '4.52.1' }
|
|
- { python: '3.9', numpy: '1.26.4', opencv: '4.7.*', torch: '1.12.0', torchvision: '0.13.0', torchaudio: '0.12.0+cpu', transformers: '4.52.1' }
|
|
- { python: '3.10', numpy: '1.26.4', opencv: '4.7.*', torch: '1.13.0', torchvision: '0.14.0', torchaudio: '0.13.0+cpu', transformers: '4.52.1' }
|
|
- { python: '3.10', numpy: '1.26.4', opencv: '4.8.*', torch: '2.0.0', torchvision: '0.15.1', torchaudio: '2.0.0+cpu', transformers: '4.52.1' }
|
|
- { python: '3.10', numpy: '1.26.4', opencv: '4.8.*', torch: '2.1.0', torchvision: '0.16.0', torchaudio: '2.1.0+cpu', transformers: '4.52.1' }
|
|
- { python: '3.11', numpy: '1.26.4', opencv: '4.9.*', torch: '2.2.1', torchvision: '0.17.1', torchaudio: '2.2.1+cpu', transformers: '4.52.1' }
|
|
- { python: '3.11', numpy: '1.26.4', opencv: '4.9.*', torch: '2.3.0', torchvision: '0.18.0', torchaudio: '2.3.0+cpu', transformers: '4.52.1' }
|
|
- { python: '3.11', numpy: '2.2.5', opencv: '4.10.*', torch: '2.4.0', torchvision: '0.19.0', torchaudio: '2.4.0+cpu', transformers: '4.52.1' }
|
|
- { python: '3.12', numpy: '2.2.5', opencv: '4.10.*', torch: '2.5.0', torchvision: '0.20.0', torchaudio: '2.5.0+cpu', transformers: '4.52.1' }
|
|
- { python: '3.12', numpy: '2.2.5', opencv: '4.11.*', torch: '2.6.0', torchvision: '0.21.0', torchaudio: '2.6.0+cpu', transformers: '4.52.1' }
|
|
- { python: '3.12', numpy: '2.2.5', opencv: '4.11.*', torch: '2.7.0', torchvision: '0.22.0', torchaudio: '2.7.0+cpu', transformers: '4.52.1' }
|
|
- { python: '3.13', numpy: '2.2.5', opencv: '4.12.*', torch: '2.8.0', torchvision: '0.23.0', torchaudio: '2.8.0+cpu', transformers: '4.56.2' }
|
|
- { python: '3.13', numpy: '2.2.5', opencv: '4.12.*', torch: '2.9.0', torchvision: '0.24.0', torchaudio: '2.9.0+cpu', transformers: '4.56.2' }
|
|
- { python: '3.13', numpy: '2.2.5', opencv: '4.12.*', torch: '2.10.0', torchvision: '0.25.0', torchaudio: '2.10.0+cpu', transformers: '4.56.2' }
|
|
- { python: '3.14', numpy: '2.3.5', opencv: '4.13.*', torch: '2.11.0', torchvision: '0.26.0', torchaudio: '2.11.0+cpu', transformers: '4.57.6' }
|
|
- { python: '3.14', numpy: '2.3.5', opencv: '4.13.*', torch: '2.12.1', torchvision: '0.27.1', torchaudio: '2.11.0+cpu', transformers: '4.57.6' }
|
|
- { python: '3.14', numpy: '2.3.5', opencv: '4.13.*', torch: '2.13.0', torchvision: '0.28.0', torchaudio: '2.11.0+cpu', transformers: '4.57.6' }
|
|
- { python: '3.14', numpy: '2.4.6', opencv: '4.14.*', torch: '2.14.0', torchvision: '0.29.0', torchaudio: '2.11.0+cpu', transformers: '5.10.4' }
|
|
|
|
name: test-${{ matrix.torch }}-py${{ matrix.python }}
|
|
|
|
env:
|
|
PYTHONUSERBASE: ${{ github.workspace }}/python-${{ matrix.python }}
|
|
|
|
steps:
|
|
- uses: actions/checkout@v7
|
|
with:
|
|
submodules: true
|
|
|
|
- name: local-cache-libtorch
|
|
id: local-cache-libtorch
|
|
uses: maxnowack/local-cache@v3
|
|
with:
|
|
path: libtorch-${{ env.LIBTORCH_VERSION }}-install
|
|
key: libtorch-${{ env.LIBTORCH_VERSION }}-linux-install-${{ env.CACHE_DATE }}
|
|
|
|
- name: local-cache-torchvision
|
|
id: local-cache-torchvision
|
|
uses: maxnowack/local-cache@v3
|
|
with:
|
|
path: torchvision-${{ env.TORCHVISION_VERSION }}-install
|
|
key: torchvision-${{ env.TORCHVISION_VERSION }}-linux-install-${{ env.CACHE_DATE }}
|
|
|
|
- name: local-cache-onnxruntime
|
|
id: local-cache-onnxruntime
|
|
uses: maxnowack/local-cache@v3
|
|
with:
|
|
path: onnxruntime-${{ env.ONNXRUNTIME_VERSION }}-install
|
|
key: onnxruntime-${{ env.ONNXRUNTIME_VERSION }}-linux-install-${{ env.CACHE_DATE }}
|
|
|
|
- name: cache-libtorch
|
|
if: steps.local-cache-libtorch.outputs.cache-hit != 'true'
|
|
id: cache-libtorch
|
|
uses: actions/cache/restore@v6
|
|
with:
|
|
path: libtorch-${{ env.LIBTORCH_VERSION }}-install
|
|
key: libtorch-${{ env.LIBTORCH_VERSION }}-linux-install-${{ env.CACHE_DATE }}
|
|
fail-on-cache-miss: true
|
|
|
|
- name: cache-torchvision
|
|
if: steps.local-cache-torchvision.outputs.cache-hit != 'true'
|
|
id: cache-torchvision
|
|
uses: actions/cache/restore@v6
|
|
with:
|
|
path: torchvision-${{ env.TORCHVISION_VERSION }}-install
|
|
key: torchvision-${{ env.TORCHVISION_VERSION }}-linux-install-${{ env.CACHE_DATE }}
|
|
fail-on-cache-miss: true
|
|
|
|
- name: cache-onnxruntime
|
|
if: steps.local-cache-onnxruntime.outputs.cache-hit != 'true'
|
|
id: cache-onnxruntime
|
|
uses: actions/cache/restore@v6
|
|
with:
|
|
path: onnxruntime-${{ env.ONNXRUNTIME_VERSION }}-install
|
|
key: onnxruntime-${{ env.ONNXRUNTIME_VERSION }}-linux-install-${{ env.CACHE_DATE }}
|
|
fail-on-cache-miss: true
|
|
|
|
- uses: actions/setup-python@v7
|
|
with:
|
|
python-version: ${{ matrix.python }}
|
|
|
|
- name: setup-pytorch
|
|
run: |
|
|
export PATH=${{ env.PYTHONUSERBASE }}/bin:$PATH
|
|
pip3 install --user pytest wheel twine requests einops numpy==${{ matrix.numpy }} opencv-python==${{ matrix.opencv }}
|
|
pip3 install --user torch==${{ matrix.torch }}+cpu torchvision==${{ matrix.torchvision }}+cpu torchaudio==${{ matrix.torchaudio }} --index-url https://download.pytorch.org/whl/cpu
|
|
pip3 install --user onnx onnxscript onnxruntime numpy==${{ matrix.numpy }}
|
|
pip3 install --user "transformers<=${{ matrix.transformers }}" diffusers "safetensors<=0.6.2"
|
|
|
|
- name: setup-pytorch-execstack-or-patchelf
|
|
if: matrix.python == '3.8' || matrix.python == '3.9'
|
|
run: |
|
|
execstack -c ${{ env.PYTHONUSERBASE }}/lib/python${{ matrix.python }}/site-packages/torch/lib/libtorch_cpu.so || true
|
|
patchelf --clear-execstack ${{ env.PYTHONUSERBASE }}/lib/python${{ matrix.python }}/site-packages/torch/lib/libtorch_cpu.so || true
|
|
|
|
- name: python-ncnn
|
|
run: |
|
|
export CMAKE_BUILD_PARALLEL_LEVEL=8
|
|
pip3 install --user . --verbose
|
|
|
|
- name: pnnx
|
|
run: |
|
|
cd tools/pnnx
|
|
mkdir build && cd build
|
|
cmake -DCMAKE_BUILD_TYPE=Release \
|
|
-DTorch_INSTALL_DIR=$GITHUB_WORKSPACE/libtorch-${{ env.LIBTORCH_VERSION }}-install \
|
|
-DTorchVision_INSTALL_DIR=$GITHUB_WORKSPACE/torchvision-${{ env.TORCHVISION_VERSION }}-install \
|
|
-Donnxruntime_INSTALL_DIR=$GITHUB_WORKSPACE/onnxruntime-${{ env.ONNXRUNTIME_VERSION }}-install \
|
|
-Dprotobuf_DIR=$GITHUB_WORKSPACE/onnxruntime-${{ env.ONNXRUNTIME_VERSION }}-install/lib/cmake/protobuf ..
|
|
|
|
- name: download-pnnx
|
|
uses: actions/download-artifact@v8
|
|
with:
|
|
name: pnnx
|
|
path: tools/pnnx/build/src
|
|
|
|
- name: test
|
|
run: |
|
|
export PATH=${{ env.PYTHONUSERBASE }}/bin:$PATH
|
|
chmod +x tools/pnnx/build/src/pnnx
|
|
export OMP_THREAD_LIMIT=1
|
|
export OMP_NUM_THREADS=1
|
|
export MKL_NUM_THREADS=1
|
|
export MKL_ENABLE_INSTRUCTIONS=SSE4_2
|
|
cd tools/pnnx/build
|
|
ctest --output-on-failure -j 8
|
|
|
|
- name: python-pnnx
|
|
run: |
|
|
export PATH=${{ env.PYTHONUSERBASE }}/bin:$PATH
|
|
export PNNX_WHEEL_WITHOUT_BUILD=ON
|
|
cd tools/pnnx/python
|
|
cp ../build/src/pnnx pnnx/
|
|
python3 setup.py install --user
|
|
pytest tests
|