PyTorch Developer Mailing List

The "Ideal" PyTorch FLOP Counter (with __torch_dispatch__)

Chillee October 20, 2022, 11:22pm 2

This is an updated version of the FLOP counter that uses TorchDispatchMode instead

1 Like

Topic		Replies	Views	Activity
What (and Why) is __torch_dispatch__? frontend API	3	11971	July 2, 2024
Estimate theoritical FLOPs of backward pass of a DNN performance	1	946	April 16, 2023
Overhead in `nn.Module` causing massive slowdowns compared to raw CuBLAS or Torchscript performance	0	1590	January 28, 2021
Torch.nn H2 2021 Lookback and H1 2022 Lookahead frontend API	0	912	January 14, 2022
NNC Per-Operator Benchmarks (on CPU) nnc	5	862	January 27, 2021