From 8cbb9d0e6ce57a8dbfc685f3121ed9d01b02726e Mon Sep 17 00:00:00 2001 From: Laurent Mazare Date: Sat, 4 Nov 2023 06:36:05 +0100 Subject: 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. --- .github/workflows/ci_cuda.yaml | 2 +- .github/workflows/rust-ci.yml | 4 ++++ 2 files changed, 5 insertions(+), 1 deletion(-) (limited to '.github') 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 -- cgit v1.2.3