search

MrYxJ / calculate-flops.pytorch

The calflops is designed to calculate FLOPs、MACs and Parameters in all various neural networks, such as Linear、 CNN、 RNN、 GCN、Transformer(Bert、LlaMA etc Large Language Model)
https://pypi.org/project/calflops/
MIT License
393 stars 14 forks source link

访存量的计算 #10

Closed Amanda-Barbara closed 5 months ago

Amanda-Barbara commented 8 months ago

这个模块中有访存量的统计计算吗？谢谢。

BitCalSaul commented 5 months ago

这个模块中有访存量的统计计算吗？谢谢。

请问您找到关于访存量的profiler吗

MrYxJ commented 5 months ago

这个工具暂时没有访存量的计算逻辑，理论上通过输入前向传播遍历一遍网络结构访存量也是可以计算出来，感谢您的提醒，之后考虑加上。