feat: export reactant compiled functions as tf SavedModel #4155
This file contains hidden or 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: Downgrade | |
| on: | |
| pull_request: | |
| branches: | |
| - main | |
| paths: | |
| - '.github/workflows/downgrade.yml' | |
| - 'ext/**' | |
| - 'lib/**' | |
| - 'src/**' | |
| - 'Project.toml' | |
| push: | |
| branches: | |
| - main | |
| paths: | |
| - '.github/workflows/downgrade.yml' | |
| - 'ext/**' | |
| - 'lib/**' | |
| - 'src/**' | |
| - 'Project.toml' | |
| concurrency: | |
| # Skip intermediate builds: always. | |
| # Cancel intermediate builds: only if it is a pull request build. | |
| group: ${{ github.workflow }}-${{ github.ref }} | |
| cancel-in-progress: ${{ startsWith(github.ref, 'refs/pull/') }} | |
| jobs: | |
| downgrade: | |
| # if: ${{ !contains(github.event.head_commit.message, '[skip tests]') && github.base_ref == github.event.repository.default_branch }} | |
| timeout-minutes: 90 | |
| runs-on: ubuntu-latest | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| test_group: | |
| - core | |
| - neural_networks | |
| - integration | |
| runtime: | |
| - PJRT | |
| - IFRT | |
| env: | |
| TMPDIR: ${{ github.workspace }}/tmp | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - name: Create TMPDIR | |
| run: | | |
| mkdir -p ${{ env.TMPDIR }} | |
| - uses: julia-actions/setup-julia@v2 | |
| with: | |
| version: "1.10" | |
| - uses: julia-actions/cache@v2 | |
| - uses: julia-actions/julia-downgrade-compat@v1 | |
| with: | |
| skip: "ReactantCore" | |
| - name: "Setup Runtime Preferences" | |
| uses: "DamianReeves/write-file-action@master" | |
| with: | |
| path: "LocalPreferences.toml" | |
| write-mode: "overwrite" | |
| contents: | | |
| [Reactant] | |
| xla_runtime = "${{ matrix.runtime }}" | |
| - name: "Install Dependencies and Run Tests" | |
| run: | | |
| import Pkg | |
| Pkg.Registry.update() | |
| # Install packages present in subdirectories | |
| dev_pks = Pkg.PackageSpec[] | |
| for path in ("lib/ReactantCore",) | |
| push!(dev_pks, Pkg.PackageSpec(; path)) | |
| end | |
| Pkg.develop(dev_pks) | |
| Pkg.test(; coverage="user") | |
| shell: julia --color=yes --code-coverage=user --depwarn=yes --project=. {0} | |
| id: run_tests | |
| env: | |
| JULIA_PKG_SERVER_REGISTRY_PREFERENCE: eager | |
| REACTANT_TEST_GROUP: ${{ matrix.test_group }} | |
| XLA_FLAGS: "--xla_force_host_platform_device_count=12" | |
| JULIA_DEBUG: "Reactant,Reactant_jll" | |
| - name: Upload MLIR modules | |
| uses: actions/upload-artifact@v4 | |
| timeout-minutes: 10 | |
| if: ${{ always() }} | |
| with: | |
| name: "mlir-downgrade-${{ matrix.test_group }}-${{ matrix.runtime }}-${{ github.event_name }}" | |
| path: "**/*.mlir" | |
| retention-days: 90 | |
| overwrite: false | |
| - uses: julia-actions/julia-processcoverage@v1 | |
| - uses: codecov/codecov-action@v5 | |
| with: | |
| files: lcov.info |