wonnx-cli 0.5.1

CLI for WONNX. WONNX is an ONNX runtime based on wgpu aimed at being a universal GPU runtime, written in Rust.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
use assert_cmd::prelude::*;
use std::{path::Path, process::Command};

#[test]
fn simple_inference() -> Result<(), Box<dyn std::error::Error>> {
    let mut cmd = Command::cargo_bin("nnx")?;

    let model_path = Path::new(env!("CARGO_MANIFEST_DIR")).join("../data/models/single_relu.onnx");

    cmd.arg("infer")
        .arg(model_path)
        .arg("-r")
        .arg("x=99,-99")
        .assert()
        .success();
    Ok(())
}