diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-11-04 06:36:05 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-11-04 06:36:05 +0100 |
| commit | 8cbb9d0e6ce57a8dbfc685f3121ed9d01b02726e (patch) | |
| tree | a49ad8154b547caa83065089bbca9066d981f03e /.github | |
| parent | bfe95115c6c55f90a4aa8712664259b5623e2935 (diff) | |
| download | candle-8cbb9d0e6ce57a8dbfc685f3121ed9d01b02726e.tar.gz candle-8cbb9d0e6ce57a8dbfc685f3121ed9d01b02726e.tar.bz2 candle-8cbb9d0e6ce57a8dbfc685f3121ed9d01b02726e.zip | |
Add some preliminary ONNX support (#1260)
* Add the onnx protos.
* Move the reading bits.
* Install protoc on the CI.
* Install protoc on the cuda CI too.
* Use clap for the onnx tool.
* Tweak the CI protoc install.
* Add some simple evalution function.
* Add some binary operator support.
Diffstat (limited to '.github')
| -rw-r--r-- | .github/workflows/ci_cuda.yaml | 2 | ||||
| -rw-r--r-- | .github/workflows/rust-ci.yml | 4 |
2 files changed, 5 insertions, 1 deletions
diff --git a/.github/workflows/ci_cuda.yaml b/.github/workflows/ci_cuda.yaml index 8953c444..ec792a25 100644 --- a/.github/workflows/ci_cuda.yaml +++ b/.github/workflows/ci_cuda.yaml @@ -59,7 +59,7 @@ jobs: - name: Install Rust Stable run: curl https://sh.rustup.rs -sSf | sh -s -- -y - uses: Swatinem/rust-cache@v2 - - run: apt-get update -y && apt-get install libssl-dev -y + - run: apt-get update -y && apt-get install libssl-dev protobuf-compiler -y - name: Test (cuda) run: PATH=$PATH:/usr/local/cuda-11.8/bin/ /root/.cargo/bin/cargo test --features cuda stop-runner: diff --git a/.github/workflows/rust-ci.yml b/.github/workflows/rust-ci.yml index 2ca53b23..b435bdfa 100644 --- a/.github/workflows/rust-ci.yml +++ b/.github/workflows/rust-ci.yml @@ -16,6 +16,7 @@ jobs: rust: [stable] steps: - uses: actions/checkout@v2 + - uses: arduino/setup-protoc@v2 - uses: actions-rs/toolchain@v1 with: profile: minimal @@ -35,6 +36,7 @@ jobs: rust: [stable] steps: - uses: actions/checkout@v2 + - uses: arduino/setup-protoc@v2 - uses: actions-rs/toolchain@v1 with: profile: minimal @@ -50,6 +52,7 @@ jobs: runs-on: ubuntu-latest steps: - uses: actions/checkout@v2 + - uses: arduino/setup-protoc@v2 - uses: actions-rs/toolchain@v1 with: profile: minimal @@ -66,6 +69,7 @@ jobs: runs-on: ubuntu-latest steps: - uses: actions/checkout@v2 + - uses: arduino/setup-protoc@v2 - uses: actions-rs/toolchain@v1 with: profile: minimal |