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

CLA assistant check
Thank you for your submission, we really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.
You have signed the CLA already but the status is still pending? Let us recheck it.

@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