Skip to content
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

pnnx print flops memops count #5648

Open
wants to merge 11 commits into
base: master
Choose a base branch
from
Open

Conversation

luxincn
Copy link
Contributor

@luxincn luxincn commented Aug 24, 2024

qwq

@tencent-adm
Copy link
Member

tencent-adm commented Aug 24, 2024

CLA assistant check
All committers have signed the CLA.

@luxincn
Copy link
Contributor Author

luxincn commented Oct 12, 2024

包含了大部分pnnx支持的op(torch.nn和torch.nn.functional),如conv,Linear,MultiheadAttention,Pool,LSTM,RNN,Mish等,并进行了简单的测试。
参考了pytorch官方文档定义 (https://pytorch.org/docs/stable/index.html) ,以及其他计算模型FLOPs和内存读取数的库,如
torchstat (https://github.com/Swall0w/torchstat) ,ptflops (https://github.com/sovrasov/flops-counter.pytorch) ,thop (https://github.com/Lyken17/pytorch-OpCounter) 。

@luxincn luxincn changed the title [WIP] pnnx print flops memops count pnnx print flops memops count Oct 13, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants