PyTorch Developer Mailing List

The "Ideal" PyTorch FLOP Counter (with __torch_dispatch__)

jeromeku August 22, 2024, 6:40pm 23

Thanks! This is super-helpful.

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