Add CLI option to measure time breakdown in GEMM bench
WilliamTambellini opened this issue · comments
William Tambellini commented
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
Benson Ma commented
Hi @WilliamTambellini sure, please feel free to open a PR for us to review
William Tambellini commented
done in
#1725