More operands for problem.yaml

AsicDyc opened this issue 5 months ago · 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.

Answer 1 · 2024-03-26T16:19:17.000Z

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.

Answer 2 · 2024-03-27T09:06:43.000Z

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.

Answer 3 · 2024-03-27T14:32:48.000Z

See if this answers your question: #22

Answer 4 · 2024-03-28T04:11:12.000Z

Thanks a lot! It helps me.