diff options
| author | Lukas Kreussel <65088241+LLukas22@users.noreply.github.com> | 2023-11-08 06:37:50 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-11-08 06:37:50 +0100 |
| commit | f3a4f3db768d46defc16de48208107db1b32159d (patch) | |
| tree | 21ae0872e46621656559ec0caf6d7625e6ca7e76 /.github | |
| parent | 7920b45c8ac737b67e23f04297f6bd7e4860f373 (diff) | |
| download | candle-f3a4f3db768d46defc16de48208107db1b32159d.tar.gz candle-f3a4f3db768d46defc16de48208107db1b32159d.tar.bz2 candle-f3a4f3db768d46defc16de48208107db1b32159d.zip | |
PyO3: Add optional `candle.onnx` module (#1282)
* Start onnx integration
* Merge remote-tracking branch 'upstream/main' into feat/pyo3-onnx
* Implement ONNXModel
* `fmt`
* add `onnx` flag to python ci
* Pin `protoc` to `25.0`
* Setup `protoc` in wheel builds
* Build wheels with `onnx`
* Install `protoc` in manylinux containers
* `apt` -> `yum`
* Download `protoc` via bash script
* Back to `manylinux: auto`
* Disable `onnx` builds for linux
Diffstat (limited to '.github')
| -rw-r--r-- | .github/workflows/maturin.yml | bin | 5304 -> 6672 bytes | |||
| -rw-r--r-- | .github/workflows/python.yml | 8 |
2 files changed, 7 insertions, 1 deletions
diff --git a/.github/workflows/maturin.yml b/.github/workflows/maturin.yml Binary files differindex 1413f014..46bdb903 100644 --- a/.github/workflows/maturin.yml +++ b/.github/workflows/maturin.yml diff --git a/.github/workflows/python.yml b/.github/workflows/python.yml index bf85f5e5..be9b917e 100644 --- a/.github/workflows/python.yml +++ b/.github/workflows/python.yml @@ -39,6 +39,12 @@ jobs: path: ~/.cargo/registry key: ${{ runner.os }}-cargo-registry-${{ hashFiles('**/Cargo.lock') }} + - name: Install Protoc + uses: arduino/setup-protoc@v2 + with: + version: "25.0" + repo-token: ${{ secrets.GITHUB_TOKEN }} + - name: Install working-directory: ./candle-pyo3 run: | @@ -46,7 +52,7 @@ jobs: source .env/bin/activate pip install -U pip pip install pytest maturin black - python -m maturin develop -r + python -m maturin develop -r --features onnx - name: Check style working-directory: ./candle-pyo3 |