[Feature Enhancement] Implement a function to collect the model's execution stats. #277

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Open

Xreki wants to merge 31 commits into PaddlePaddle:develop

from Xreki:collect_info

Open

[Feature Enhancement] Implement a function to collect the model's execution stats. #277

Xreki wants to merge 31 commits into PaddlePaddle:develop from Xreki:collect_info

Conversation

@Xreki

Copy link

Collaborator

@Xreki Xreki commented Sep 16, 2025 •

edited

Loading

PR Category

Feature Enhancement

Description

执行结果如下:

[2025年09月22日 21:04:20.401] Collect information for /work/GraphNet/samples/torchvision/resnet18
Totally 1 subgraphs, 69 operators, and 0 operators failed to inference dtypes.
model_stats collection information: model_path=/work/GraphNet/samples/torchvision/resnet18, method=symbolic_trace, is_ops_complete=True
graph-net-collect-stats-log [ModelStats.num_inputs] model_path:/work/GraphNet/samples/torchvision/resnet18 42
graph-net-collect-stats-log [ModelStats.num_params] model_path:/work/GraphNet/samples/torchvision/resnet18 62
graph-net-collect-stats-log [ModelStats.num_outputs] model_path:/work/GraphNet/samples/torchvision/resnet18 1
graph-net-collect-stats-log [ModelStats.num_ops] model_path:/work/GraphNet/samples/torchvision/resnet18 69
graph-net-collect-stats-log [ModelStats.model_size] model_path:/work/GraphNet/samples/torchvision/resnet18 0.011689512B
graph-net-collect-stats-log [ModelStats.input_dtypes] model_path:/work/GraphNet/samples/torchvision/resnet18 int64:1 float32:41
graph-net-collect-stats-log [ModelStats.param_dtypes] model_path:/work/GraphNet/samples/torchvision/resnet18 float32:62
graph-net-collect-stats-log [ModelStats.op_dtypes] model_path:/work/GraphNet/samples/torchvision/resnet18 float32:69
graph-net-collect-stats-log [ModelStats.ops] model_path:/work/GraphNet/samples/torchvision/resnet18 adaptive_avg_pool2d:1 add:8 batch_norm:20 conv2d:20 flatten:1 linear:1 max_pool2d:1 relu:17
graph-net-collect-stats-log [ModelStats.source] model_path:/work/GraphNet/samples/torchvision/resnet18 torchvision
graph-net-collect-stats-log [ModelStats.heuristic_tag] model_path:/work/GraphNet/samples/torchvision/resnet18 computer_vision

@paddle-bot

Copy link

paddle-bot bot commented Sep 16, 2025

Thanks for your contribution!

Xreki added 3 commits

September 16, 2025 10:56

@Xreki


 Implement a function to collect the model's execution stats.

7a91e19

@Xreki


 Add support of get_attr and simplify some codes.

bcf9d5a

@Xreki


 Fix support of call_method.

@Xreki Xreki force-pushed the collect_info branch from 272ab55 to 1415926 Compare

September 16, 2025 07:22

@Xreki


 Support _native_multi_head_attention.

dbcadfa

@Xreki Xreki force-pushed the collect_info branch from b5bd25b to dbcadfa Compare

September 16, 2025 09:04

Xreki added 10 commits

September 16, 2025 17:22

@Xreki


 Merge branch 'develop' into collect_info

b2073f9

@Xreki


 Fix several ops and change to use subprocess for multiple tests.

8161df7

@Xreki


 Support another method with make_fx.

07558f2

@Xreki


 Optimize the dtypes stats.

256c75f

@Xreki


 Enable to print error messages.

a3fb5ae

@Xreki


 Fix several problems.

d10dcc3

@Xreki


 Support to rerun the failed cases only.

f159a3d

@Xreki


 Implement method using torch.compile with customized backend.

ee5fd22

@Xreki


 Update the log format.

777f8dd

@Xreki


 Merge branch 'develop' into collect_info

9f3086a

@Xreki Xreki force-pushed the collect_info branch from 1df5d5f to 6984169 Compare

September 18, 2025 08:50

@Xreki


 Add source and heuristic_tag.

9cf8a86

@Xreki Xreki force-pushed the collect_info branch from 6984169 to 9cf8a86 Compare

September 18, 2025 08:56

Xreki added 4 commits

September 18, 2025 19:08

@Xreki


 Add timestamp in log.

b1eb293

@Xreki


 Merge branch 'develop' into collect_info

@Xreki


 Merge branch 'develop' into collect_info

a738b6b

@Xreki


 Remove the make_fx implementation.

@Xreki Xreki force-pushed the collect_info branch from c14fed5 to 7575957 Compare

September 19, 2025 01:11

Xreki added 5 commits

September 22, 2025 16:43

@Xreki


 Add paddle implementation.

0ce778f

@Xreki


 Reorganize some codes.

1a3dc30

@Xreki


 Merge branch 'develop' into collect_info

449f38c

@Xreki


 Reorgnanize codes.

5db7117

@Xreki


 Support to collect input shapes and use json to dump list and dict.

6a5b8db

Xreki added 7 commits

September 23, 2025 22:04

@Xreki


 Merge branch 'develop' into collect_info

eb9c0ae

@Xreki


 Merge branch 'develop' into collect_info

0e8ca31

@Xreki


 Merge branch 'develop' into collect_info

6a1a2aa

@Xreki


 Merge branch 'develop' into collect_info

c6854bf

@Xreki


 Merge branch 'develop' into collect_info

6f53c99

@Xreki


 Merge branch 'develop' into collect_info

aa7d4d0

@Xreki


 Merge branch 'develop' into collect_info

d101b02

Navigation Menu

Search code, repositories, users, issues, pull requests...

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

[Feature Enhancement] Implement a function to collect the model's execution stats. #277

Are you sure you want to change the base?

[Feature Enhancement] Implement a function to collect the model's execution stats. #277

Uh oh!

Conversation

@Xreki Xreki commented Sep 16, 2025 •

edited

Loading

Uh oh!

PR Category

Description

Uh oh!

paddle-bot bot commented Sep 16, 2025

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

1 participant

[Feature Enhancement] Implement a function to collect the model's execution stats. #277

Are you sure you want to change the base?

[Feature Enhancement] Implement a function to collect the model's execution stats. #277

Uh oh!

Conversation

@Xreki Xreki commented Sep 16, 2025 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

PR Category

Description

Uh oh!

paddle-bot bot commented Sep 16, 2025

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

1 participant

@Xreki Xreki commented Sep 16, 2025 •

edited

Loading