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 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.