[RVV] Add missing maxpool and avgpool rvv kernels #9622
[RVV] Add missing maxpool and avgpool rvv kernels #9622copybara-service[bot] merged 4 commits intogoogle:masterfrom
Conversation
@dsharlet just fyi, I've opened a PR in pytorch/cpuinfo for this purpose. Once that is reviewed and merged I can make the needed changes to hardware-config.c. pytorch/cpuinfo#375 While other changes in cpuinfo are desirable (e.g cache info for riscv uarchs) I tried to keep the changes to a minimum this round in the hope that the review goes smoothly. |
|
@fbarchard please also give this PR a lookover when you have a few minutes. Thank you. |
|
@fbarchard could you review when you have a few free minutes. Thank you. |
|
@dsharletg I've updated this PR to latest master and retested. Please review and merge when you are able. Thank you! With this, subgraph-mobilenet-bench now runs with the FP16 datatype. |
|
@dsharletg thank you for all the previous PR approvals and merges. A very friendly ping on this one. I think we (or at least I) are getting very close to done. |
Relevant tests and benchmarks executed and pass (BPI-F3).