[nnx] add flaxlib #17
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Flaxlib - Build and upload to PyPI | |
| # for testing only: | |
| on: | |
| pull_request: | |
| branches: [main] | |
| # on: | |
| # workflow_dispatch: | |
| # pull_request: | |
| # push: | |
| # branches: [main] | |
| # paths: ['flaxlib/**'] | |
| # release: | |
| # types: [published] | |
| jobs: | |
| build_wheels: | |
| name: Build wheels on ${{ matrix.os }} | |
| runs-on: ${{ matrix.os }} | |
| strategy: | |
| matrix: | |
| # macos-13 is an intel runner, macos-14 is apple silicon | |
| os: [ubuntu-latest, windows-latest, macos-13, macos-14] | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - uses: actions/setup-python@v5 | |
| - name: Setup Rust | |
| uses: actions-rust-lang/setup-rust-toolchain@v1 | |
| - name: Install cibuildwheel | |
| run: python -m pip install cibuildwheel==2.21.0 | |
| - name: Build wheels | |
| run: python -m cibuildwheel --output-dir ./flaxlib/wheelhouse ./flaxlib | |
| env: | |
| # rust doesn't seem to be available for musl linux on i686 | |
| CIBW_SKIP: "*-musllinux_i686" | |
| CIBW_ENVIRONMENT: 'PATH="$HOME/.cargo/bin:$PATH" CARGO_TERM_COLOR="always"' | |
| CIBW_ENVIRONMENT_WINDOWS: 'PATH="$UserProfile\.cargo\bin;$PATH"' | |
| CIBW_BEFORE_BUILD: rustup show | |
| CIBW_BEFORE_BUILD_LINUX: | | |
| curl https://sh.rustup.rs -sSf | sh -s -- --default-toolchain=stable --profile=minimal -y && | |
| rustup show | |
| - uses: actions/upload-artifact@v4 | |
| with: | |
| name: cibw-wheels-${{ matrix.os }}-${{ strategy.job-index }} | |
| path: ./flaxlib/wheelhouse/*.whl | |
| build_sdist: | |
| name: Build source distribution | |
| runs-on: ubuntu-latest | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - name: Setup Rust | |
| uses: actions-rust-lang/setup-rust-toolchain@v1 | |
| - name: Build sdist | |
| run: pipx run build --sdist flaxlib | |
| - uses: actions/upload-artifact@v4 | |
| with: | |
| name: cibw-sdist | |
| path: ./flaxlib/dist/*.tar.gz | |
| upload_pypi: | |
| name: Upload to PyPI | |
| needs: [build_wheels, build_sdist] | |
| runs-on: ubuntu-latest | |
| permissions: | |
| id-token: write | |
| steps: | |
| - uses: actions/setup-python@v1 | |
| with: | |
| python-version: '3.x' | |
| - name: Install dependencies | |
| run: | | |
| python -m pip install --upgrade pip | |
| pip install setuptools build wheel twine | |
| - uses: actions/download-artifact@v4 | |
| with: | |
| # unpacks all CIBW artifacts into dist/ | |
| pattern: cibw-* | |
| path: ./flaxlib/dist | |
| merge-multiple: true | |
| - name: Build and publish | |
| env: | |
| TWINE_USERNAME: __token__ | |
| TWINE_PASSWORD: ${{ secrets.FLAXLIB_PYPI_TOKEN }} | |
| run: | | |
| twine upload flaxlib/dist/* |