Add some CLI args to set M,N,K to GEMMsBenchmark
WilliamTambellini opened this issue · comments
Hi,
Would you mind if I add few CLI args to set M, N and/or K for GEMMsBenchmark:
https://github.com/pytorch/FBGEMM/blob/main/bench/GEMMsBenchmark.cc
eg:
./GEMMsBenchmark -M 13263 -N 242 -K 2821
?
Hi @WilliamTambellini yes, absolutely, please feel free to open a PR that we can review.
closed by
#1721