Skip to content

Commit

Permalink
Reland #94719 - Update ideep to add primitive cache for ARM (#95688)
Browse files Browse the repository at this point in the history 
### Description
This PR is to update ideep to add primitive cache in order to speed up ARM's PyTorch workloads.
Reland pytorch/pytorch#94719, which is unintentional reverted by pytorch/pytorch#94939 (comment).
Fixes pytorch/pytorch#94264.

### Performance test
Use TorchBench test in ICX with 40 cores
Intel OpenMP & jemalloc were preloaded
![image](https://user-images.githubusercontent.com/61222868/221760391-fb6cbabe-6d88-4155-b216-348e718e68b9.png)

Pull Request resolved: pytorch/pytorch#95688
Approved by: https://github.com/ezyang
  • Loading branch information
@yanbing-j @cyyever
yanbing-j authored and cyyever committed Mar 5, 2023
1 parent 3cad0c6 commit 8d71542
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion third_party/ideep
Submodule ideep updated 6 files
+9 −1 include/ideep/lru_cache.hpp
+186 −0 include/ideep/operators/conv.hpp
+76 −0 include/ideep/operators/inner_product.hpp
+54 −0 include/ideep/operators/matmul.hpp
+12 −0 include/ideep/tensor.hpp
+44 −0 include/ideep/utils.hpp

0 comments on commit 8d71542

Please sign in to comment.