From f3a4f3db768d46defc16de48208107db1b32159d Mon Sep 17 00:00:00 2001
From: Lukas Kreussel <65088241+LLukas22@users.noreply.github.com>
Date: Wed, 8 Nov 2023 06:37:50 +0100
Subject: 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
---
 .github/workflows/maturin.yml | Bin 5304 -> 6672 bytes
 .github/workflows/python.yml  |   8 +++++++-
 2 files changed, 7 insertions(+), 1 deletion(-)

(limited to '.github')

diff --git a/.github/workflows/maturin.yml b/.github/workflows/maturin.yml
index 1413f014..46bdb903 100644
Binary files a/.github/workflows/maturin.yml and b/.github/workflows/maturin.yml differ
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
-- 
cgit v1.2.3