pytorch / FBGEMM

FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Add CLI option to measure time breakdown in GEMM bench

WilliamTambellini opened this issue · comments

In
https://github.com/pytorch/FBGEMM/blob/main/bench/GEMMsBenchmark.cc
there is a define in order to measure and log the time breakdown.
Would you mind if I convert it to a runtime cli arg ?
eg
./GEMMsBenchmark --timebreakdown=1
default 0

Hi @WilliamTambellini sure, please feel free to open a PR for us to review