More operands for problem.yaml
AsicDyc opened this issue · 4 comments
Hi,
Thank you for providing such a useful tool.
Just want to know does timeloop support other operands like matrix multiplication or depth-wise convolution? Are there any tutorials or examples for these operands?
Thanks for your assistance.
Yes. GEMM is already provided as a problem shape in https://github.com/NVlabs/timeloop/blob/master/problem-shapes/gemm_ABZ.yaml. Please watch our video tutorials in order to understand how to describe problem shapes.
Thanks a lot for your prompt reply. What about depth-wise convolution? The calculation methods between depth-wise convolution and traditional convolution differ, so simply describing depth-wise convolution as a cnn_layer type in prob.yaml would lead to issues.
See if this answers your question: #22
Thanks a lot! It helps me.