dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
burn/crates/burn-import/onnx-tests/build.rs

199 lines
7.9 KiB
Rust
Raw Permalink Blame History

use burn_import::onnx::{ModelGen, RecordType};
fn main() {
// Re-run this build script if the onnx-tests directory changes.
println!("cargo:rerun-if-changed=tests");
// Add onnx models.
ModelGen::new()
.input("tests/add/add.onnx")
.input("tests/add/add_int.onnx")
.input("tests/argmax/argmax.onnx")
.input("tests/avg_pool1d/avg_pool1d.onnx")
.input("tests/avg_pool2d/avg_pool2d.onnx")
.input("tests/batch_norm/batch_norm.onnx")
.input("tests/cast/cast.onnx")
.input("tests/clip/clip_opset16.onnx")
.input("tests/clip/clip_opset7.onnx")
.input("tests/concat/concat.onnx")
.input("tests/constant_of_shape/constant_of_shape.onnx")
.input("tests/constant_of_shape/constant_of_shape_full_like.onnx")
.input("tests/conv1d/conv1d.onnx")
.input("tests/conv2d/conv2d.onnx")
.input("tests/conv3d/conv3d.onnx")
.input("tests/conv_transpose1d/conv_transpose1d.onnx")
.input("tests/conv_transpose2d/conv_transpose2d.onnx")
.input("tests/conv_transpose3d/conv_transpose3d.onnx")
.input("tests/cos/cos.onnx")
.input("tests/div/div.onnx")
.input("tests/dropout/dropout_opset16.onnx")
.input("tests/dropout/dropout_opset7.onnx")
.input("tests/equal/equal.onnx")
.input("tests/erf/erf.onnx")
.input("tests/exp/exp.onnx")
.input("tests/expand/expand.onnx")
.input("tests/expand/expand_tensor.onnx")
.input("tests/expand/expand_shape.onnx")
.input("tests/flatten/flatten.onnx")
.input("tests/floor/floor.onnx")
.input("tests/gather/gather_1d_idx.onnx")
.input("tests/gather/gather_2d_idx.onnx")
.input("tests/gather/gather_scalar.onnx")
.input("tests/gather/gather_shape.onnx")
.input("tests/gather_elements/gather_elements.onnx")
.input("tests/gelu/gelu.onnx")
.input("tests/global_avr_pool/global_avr_pool.onnx")
.input("tests/greater/greater.onnx")
.input("tests/greater/greater_scalar.onnx")
.input("tests/greater_or_equal/greater_or_equal.onnx")
.input("tests/greater_or_equal/greater_or_equal_scalar.onnx")
.input("tests/hard_sigmoid/hard_sigmoid.onnx")
.input("tests/layer_norm/layer_norm.onnx")
.input("tests/leaky_relu/leaky_relu.onnx")
.input("tests/less/less.onnx")
.input("tests/less/less_scalar.onnx")
.input("tests/less_or_equal/less_or_equal.onnx")
.input("tests/less_or_equal/less_or_equal_scalar.onnx")
.input("tests/linear/linear.onnx")
.input("tests/log/log.onnx")
.input("tests/log_softmax/log_softmax.onnx")
.input("tests/mask_where/mask_where.onnx")
.input("tests/mask_where/mask_where_broadcast.onnx")
.input("tests/mask_where/mask_where_scalar_x.onnx")
.input("tests/mask_where/mask_where_scalar_y.onnx")
.input("tests/mask_where/mask_where_all_scalar.onnx")
.input("tests/matmul/matmul.onnx")
.input("tests/max/max.onnx")
.input("tests/maxpool1d/maxpool1d.onnx")
.input("tests/maxpool2d/maxpool2d.onnx")
.input("tests/min/min.onnx")
.input("tests/mean/mean.onnx")
.input("tests/mul/mul.onnx")
.input("tests/neg/neg.onnx")
.input("tests/not/not.onnx")
.input("tests/one_hot/one_hot.onnx")
.input("tests/pad/pad.onnx")
.input("tests/pow/pow.onnx")
.input("tests/pow/pow_int.onnx")
.input("tests/prelu/prelu.onnx")
.input("tests/random_normal/random_normal.onnx")
.input("tests/random_normal_like/random_normal_like.onnx")
.input("tests/random_uniform/random_uniform.onnx")
.input("tests/random_uniform_like/random_uniform_like.onnx")
.input("tests/range/range.onnx")
.input("tests/recip/recip.onnx")
.input("tests/reduce_max/reduce_max.onnx")
.input("tests/reduce_mean/reduce_mean.onnx")
.input("tests/reduce_min/reduce_min.onnx")
.input("tests/reduce_prod/reduce_prod.onnx")
.input("tests/reduce_sum/reduce_sum_opset11.onnx")
.input("tests/reduce_sum/reduce_sum_opset13.onnx")
.input("tests/relu/relu.onnx")
.input("tests/reshape/reshape.onnx")
.input("tests/resize/resize_with_sizes.onnx")
.input("tests/resize/resize_1d_linear_scale.onnx")
.input("tests/resize/resize_1d_nearest_scale.onnx")
.input("tests/resize/resize_2d_bicubic_scale.onnx")
.input("tests/resize/resize_2d_bilinear_scale.onnx")
.input("tests/resize/resize_2d_nearest_scale.onnx")
.input("tests/shape/shape.onnx")
.input("tests/sigmoid/sigmoid.onnx")
.input("tests/sign/sign.onnx")
.input("tests/sin/sin.onnx")
.input("tests/slice/slice.onnx")
.input("tests/softmax/softmax.onnx")
.input("tests/sqrt/sqrt.onnx")
.input("tests/squeeze/squeeze_multiple.onnx")
.input("tests/squeeze/squeeze_opset13.onnx")
.input("tests/squeeze/squeeze_opset16.onnx")
.input("tests/sub/sub.onnx")
.input("tests/sub/sub_int.onnx")
.input("tests/sum/sum.onnx")
.input("tests/sum/sum_int.onnx")
.input("tests/tanh/tanh.onnx")
.input("tests/tile/tile.onnx")
.input("tests/top_k/top_k_opset_1.onnx")
.input("tests/trilu/trilu_upper.onnx")
.input("tests/trilu/trilu_lower.onnx")
.input("tests/transpose/transpose.onnx")
.input("tests/unsqueeze/unsqueeze.onnx")
.input("tests/unsqueeze/unsqueeze_opset11.onnx")
.input("tests/unsqueeze/unsqueeze_opset16.onnx")
.input("tests/split/split.onnx")
.out_dir("model/")
.run_from_script();
// The following tests are used to generate the model with different record types.
// (e.g. bincode, pretty_json, etc.) Do not need to add new tests here, just use the default
// record type to the ModelGen::new() call above.
ModelGen::new()
.input("tests/conv1d/conv1d.onnx")
.out_dir("model/named_mpk/")
.record_type(RecordType::NamedMpk)
.run_from_script();
ModelGen::new()
.input("tests/conv1d/conv1d.onnx")
.out_dir("model/named_mpk_half/")
.record_type(RecordType::NamedMpk)
.half_precision(true)
.run_from_script();
ModelGen::new()
.input("tests/conv1d/conv1d.onnx")
.out_dir("model/pretty_json/")
.record_type(RecordType::PrettyJson)
.run_from_script();
ModelGen::new()
.input("tests/conv1d/conv1d.onnx")
.out_dir("model/pretty_json_half/")
.record_type(RecordType::PrettyJson)
.half_precision(true)
.run_from_script();
ModelGen::new()
.input("tests/conv1d/conv1d.onnx")
.out_dir("model/named_mpk_gz/")
.record_type(RecordType::NamedMpkGz)
.run_from_script();
ModelGen::new()
.input("tests/conv1d/conv1d.onnx")
.out_dir("model/named_mpk_gz_half/")
.record_type(RecordType::NamedMpkGz)
.half_precision(true)
.run_from_script();
ModelGen::new()
.input("tests/conv1d/conv1d.onnx")
.out_dir("model/bincode/")
.record_type(RecordType::Bincode)
.run_from_script();
ModelGen::new()
.input("tests/conv1d/conv1d.onnx")
.out_dir("model/bincode_half/")
.record_type(RecordType::Bincode)
.half_precision(true)
.run_from_script();
ModelGen::new()
.input("tests/conv1d/conv1d.onnx")
.out_dir("model/bincode_embedded/")
.embed_states(true)
.record_type(RecordType::Bincode)
.run_from_script();
ModelGen::new()
.input("tests/conv1d/conv1d.onnx")
.out_dir("model/bincode_embedded_half/")
.embed_states(true)
.half_precision(true)
.record_type(RecordType::Bincode)
.run_from_script();
// panic!("Purposefully failing build to output logs.");
}
Reference in New Issue View Git Blame Copy Permalink