search

NVlabs / timeloop

Timeloop performs modeling, mapping and code-generation for tensor algebra workloads on various accelerator architectures.
https://timeloop.csail.mit.edu/
BSD 3-Clause "New" or "Revised" License
325 stars 101 forks source link

More operands for problem.yaml #251

Closed AsicDyc closed 6 months ago

AsicDyc commented 6 months ago

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.

angshuman-parashar commented 6 months ago

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.

AsicDyc commented 6 months ago

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.

angshuman-parashar commented 6 months ago

See if this answers your question: https://github.com/NVlabs/timeloop/issues/22

AsicDyc commented 6 months ago

Thanks a lot! It helps me.