Make `nnx info` deterministic

webonnx / wonnx

A WebGPU-accelerated ONNX inference run-time written 100% in Rust, ready for native and the web

Other

1.65k stars 60 forks source link

Make `nnx info` deterministic #90

Closed SludgePhD closed 2 years ago

SludgePhD commented 2 years ago

The use of HashMap and HashSet made the output of nnx info <model> nondeterministic, which makes it hard to compare it between an unoptimized and optimized version of the same model. This PR switches the relevant data structures to their deterministic BTree variants to fix that.

pixelspark commented 2 years ago

Thanks :+1: looks good to me! The CI fail is unrelated to your PR, so will merge.