Skip to content

[tools/onnx-subgraph] add config.json for onnx model splitting#14852

Merged
seanshpark merged 2 commits into
Samsung:masterfrom
chenyx113:onnx-subgraph-0313
Mar 18, 2025
Merged

[tools/onnx-subgraph] add config.json for onnx model splitting#14852
seanshpark merged 2 commits into
Samsung:masterfrom
chenyx113:onnx-subgraph-0313

Conversation

@chenyx113

Copy link
Copy Markdown
Contributor

related issue: #14534
historical full changes PR: #14613

config.json for onnx model splitting, it will be used in future PRs

ONE-DCO-1.0-Signed-off-by: Youxin Chen yx113.chen@samsung.com

config.json for onnx model splitting, it will be used in future PRs

ONE-DCO-1.0-Signed-off-by: Youxin Chen <yx113.chen@samsung.com>
@chenyx113 chenyx113 marked this pull request as ready for review March 14, 2025 13:35
Comment thread tools/onnx_subgraph/config.json Outdated
Comment on lines +2 to +4
"NPU_supported_ops": ["Conv", "Reshape", "Transpose", "Add", "ReduceMean", "Sub", "Div", "Mul", "Sigmoid","MatMul"],
"CPU_supported_ops": ["Sub", "Pow", "ReduceMean", "Add", "Sqrt", "Div","Transpose", "Gather", "MatMul", "Mul", "Softmax", "Erf", "Gemm", "Conv", "Reshape",
"Sin", "Where", "ConstantOfShape", "Cast", "Sigmoid", "Cos", "Expand", "Slice", "Unsqueeze"],

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

it would be nice if you could split lines so that text comes inside 100 cols.

Comment thread tools/onnx_subgraph/config.json Outdated
Comment on lines +6 to +7
{"name":"Conv","CPU_time": 0.1, "NPU_time": 0.05},
{"name":"Mul", "CPU_time": 0.15, "NPU_time": 0.07}

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

As I understand, if the performance data is changed by targets, we should update these values.
update to these won't be good with source code management.
So, I suggest not to put these in here but to explain in the README file to add in performance_data item.
And/Or you can provide something like config.sample.json file with explanation in README so that users can follow by an example.

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

yes, I agree with you, I add it to examples path, and will add description for this config in readme later, thank you

Comment thread tools/onnx_subgraph/config.json Outdated
{"name":"Conv","CPU_time": 0.1, "NPU_time": 0.05},
{"name":"Mul", "CPU_time": 0.15, "NPU_time": 0.07}
],
"hardware_limits": {

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

ditto

@seanshpark seanshpark left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@seanshpark seanshpark merged commit e5e151d into Samsung:master Mar 18, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants