saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-23 22:13:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

lint

Browse source
This commit is contained in:
Jingyan Wang 2025-02-05 22:25:34 +00:00
parent 8fc043f474
commit 24cde8cabe
2 changed files with 3 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
onnxruntime/test/providers/tensorrt/tensorrt_basic_test.cc
View file

@ -209,7 +209,6 @@ void CreateParititionedModel(const PathString& model_name,
status = onnxruntime::Model::Save(model, model_name);
}
std::vector<char> ReadFileFromDisk(const PathString& path) {
std::fstream file(path.c_str(), std::fstream::binary | std::fstream::in | std::fstream::ate);
std::vector<char> file_bytes;
@ -475,7 +474,7 @@ TEST(TensorrtExecutionProviderTest, EPContextNode) {
std::string ctx_model_path = "EP_Context_model.onnx";
std::string sess_log_id = "EPContextNode_test";
std::vector<int> dims = {1, 3, 2};
remove(ctx_model_path.c_str()); // remove the context model file generated by previous test
CreateBaseModel(model_name, graph_name, dims);
@ -774,7 +773,7 @@ TEST(TensorrtExecutionProviderTest, EPContextNodeMulti) {
// prepare expected inputs and outputs
std::vector<int64_t> expected_dims_mul_m = {3, 6};
std::vector<int64_t> expected_values_mul_m = { 1, 2, 3, 4, 5, 6, 7, 8, 10, 11, 13, 14, 14, 16, 16, 18, 0, 1 };
std::vector<int64_t> expected_values_mul_m = {1, 2, 3, 4, 5, 6, 7, 8, 10, 11, 13, 14, 14, 16, 16, 18, 0, 1};
/*
* Test case 1.1: Dump context model to current directory, context saved in engine cache

5
onnxruntime/test/python/onnxruntime_test_python_nested_control_flow_op.py
View file

@ -205,15 +205,12 @@ def make_opt_nested_greater_or_equal() -> ModelProto:
m = helper.make_model(
graph,
opset_imports=[
helper.make_opsetid("", 17),
# helper.make_opsetid("", 15),
helper.make_opsetid("", 15),
],
)
checker.check_model(m, full_check=True)
# onnx.save_model(m, "nested_control_flow_model_opset17.onnx")
return m