yuchaoli opened a new pull request #8065: URL: https://github.com/apache/tvm/pull/8065
This pull request adds sparse conv2d(1*1) support for auto_scheduler. I have implemented sparse conv2d with kernel_size=1, using the BSR sparse data format. I evaluate the performance of sparse conv2d by mobilenet_v2 in pixel2. All of 1*1 conv2d in the network are transformed into sparse conv2d with block_size=(4, 1). | Sparse Rate | 0 %| 10 %| 20 %| 30 %| 40 %| 50 %| 60 %| 70 %| 80 %| 90 %| |---------|-------|-------|-------|-------|-------|-------|-------|-------|-------|-------| | Latency | 17.37ms| 21.85ms | 21.05ms | 18.18ms | 17.16ms | 15.03ms | 14.06ms | 11.59ms | 10.12ms | 8.67ms | @jcf94 -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected]
