apr-cli 0.4.15

CLI tool for APR model inspection, debugging, and operations
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80


// ============================================================================
// Output Functions
// ============================================================================

fn output_json(report: &DiffReport) {
    let json_result = DiffResultJson::from(report);
    if let Ok(json) = serde_json::to_string_pretty(&json_result) {
        println!("{json}");
    }
}

fn output_text(report: &DiffReport, show_weights: bool) {
    output::header("Model Diff");

    let format_info = if report.same_format() {
        report.format1.clone()
    } else {
        format!("{} vs {}", report.format1, report.format2)
    };

    println!(
        "{}",
        output::kv_table(&[
            ("File A", report.path1.clone()),
            ("File B", report.path2.clone()),
            ("Format", format_info),
        ])
    );
    println!();

    if report.is_identical() {
        println!(
            "  {}",
            output::badge_pass("Models are IDENTICAL in structure and metadata")
        );
    } else {
        let count = report.diff_count();
        println!(
            "  {} {} differences found",
            output::badge_warn("DIFF"),
            count
        );
        println!();

        // Build diff table
        let mut rows: Vec<Vec<String>> = Vec::new();
        for category in [
            DiffCategory::Format,
            DiffCategory::Size,
            DiffCategory::Quantization,
            DiffCategory::Metadata,
            DiffCategory::Tensor,
        ] {
            let diffs = report.differences_by_category(category);
            for diff in diffs {
                rows.push(vec![
                    category.name().to_string(),
                    diff.field.clone(),
                    diff.value1.clone(),
                    diff.value2.clone(),
                ]);
            }
        }
        if !rows.is_empty() {
            println!(
                "{}",
                output::table(&["Category", "Field", "File A", "File B"], &rows)
            );
        }
    }

    if show_weights {
        println!();
        println!(
            "  {} Use --values to compare actual tensor values",
            output::badge_info("TIP")
        );
    }
}