Hui Wu
31ab03e34f
Add Winograd Conv method for CPU ( #15196 )
...
Summary:
Add winograd conv method. Users can select the direct conv or winograd conv in the model file.
We close the origin pr https://github.com/pytorch/pytorch/pull/12154 and create this new one for better rebasing.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15196
Differential Revision: D13463721
Pulled By: yinghai
fbshipit-source-id: c5cd5c8aa7622ae7e52aeabd3dbb8ffb99b9b4ee
2019-02-01 16:41:30 -08:00
Gu, Jinghui
dbab9b73b6
seperate mkl, mklml, and mkldnn ( #12170 )
...
Summary:
1. Remove avx2 support in mkldnn
2. Seperate mkl, mklml, and mkldnn
3. Fix convfusion test case
Pull Request resolved: https://github.com/pytorch/pytorch/pull/12170
Reviewed By: yinghai
Differential Revision: D10207126
Pulled By: orionr
fbshipit-source-id: 1e62eb47943f426a89d57e2d2606439f2b04fd51
2018-10-29 10:52:55 -07:00
Gu, Jinghui
e8b8c3895e
Enable Conv fusion optimizations in optimizeForIdeep ( #9255 )
...
Summary:
Enable fusion for IDEEP in optimizeForIdeep
including Conv+ReLU, Conv+Sum, Conv+Sum+ReLU, Conv+BN
Pull Request resolved: https://github.com/pytorch/pytorch/pull/9255
Reviewed By: bddppq
Differential Revision: D8809030
Pulled By: yinghai
fbshipit-source-id: af30bad3b96cb965bd26a4dfa810370faec4bb88
2018-07-16 21:28:50 -07:00
Yinghai Lu
2ed03898cd
Add depthwise convolution test for IDEEP ( #8301 )
2018-06-09 08:44:13 -07:00
Sebastian Meßmer
b3e87b1066
Fix fbcode compatibility ( #7939 )
2018-05-30 13:35:46 -04:00
Yinghai Lu
2863d935b9
[Caffe2] Fix of the performance issue of IDEEP ( #7503 )
...
* Sketch fix of the performance issue of IDEEP
* Revert CMakefile
* Fix tests
* format
* comments
* Print error
* review comments
2018-05-11 13:43:41 -07:00
Jinghui
769397eb77
[Caffe2] [feature request] Add gradient operators for IDEEP ( #7234 )
...
* Add gradient operators for IDEEP
Signed-off-by: Gu, Jinghui <jinghui.gu@intel.com>
* Add gradient test cases for IDEEP
Signed-off-by: Gu, Jinghui <jinghui.gu@intel.com>
* Upgrade third_party/ideep
Signed-off-by: Gu, Jinghui <jinghui.gu@intel.com>
* Refine SumOp for IDEEP
Signed-off-by: Gu, Jinghui <jinghui.gu@intel.com>
* Share input buffer in fallback op if possible
Signed-off-by: Gu, Jinghui <jinghui.gu@intel.com>
* Fallback ConvTranspose op for IDEEP
Signed-off-by: Gu, Jinghui <jinghui.gu@intel.com>
* Fix bug introduced by the patch of sharing input buffer
Signed-off-by: Gu, Jinghui <jinghui.gu@intel.com>
* Share output buffer in fallback operators
Signed-off-by: Gu, Jinghui <jinghui.gu@intel.com>
* Remove IDEEP to resolve repo issue
Signed-off-by: Gu, Jinghui <jinghui.gu@intel.com>
* Reflash IDEEP repo
Signed-off-by: Gu, Jinghui <jinghui.gu@intel.com>
* Remove redundant lines in IDEEP
Signed-off-by: Gu, Jinghui <jinghui.gu@intel.com>
* Fallback operators for IDEEP
(Flatten, ResizeLike, Transpose, and Reshape)
Signed-off-by: Gu, Jinghui <jinghui.gu@intel.com>
2018-05-09 08:52:24 -07:00
Jinghui
26ddefbda1
[feature request] [Caffe2] Enable MKLDNN support for inference ( #6699 )
...
* Add operators based-on IDEEP interfaces
Signed-off-by: Gu, Jinghui <jinghui.gu@intel.com>
* Enable IDEEP as a caffe2 device
Signed-off-by: Gu, Jinghui <jinghui.gu@intel.com>
* Add test cases for IDEEP ops
Signed-off-by: Gu, Jinghui <jinghui.gu@intel.com>
* Add IDEEP as a caffe2 submodule
Signed-off-by: Gu, Jinghui <jinghui.gu@intel.com>
* Skip test cases if no IDEEP support
Signed-off-by: Gu, Jinghui <jinghui.gu@intel.com>
* Correct cmake options for IDEEP
Signed-off-by: Gu, Jinghui <jinghui.gu@intel.com>
* Add dependences on ideep libraries
Signed-off-by: Gu, Jinghui <jinghui.gu@intel.com>
* Fix issues in IDEEP conv ops and etc.
Signed-off-by: Gu, Jinghui <jinghui.gu@intel.com>
* Move ideep from caffe2/ideep to caffe2/contrib/ideep
Signed-off-by: Gu Jinghui <jinghui.gu@intel.com>
* Update IDEEP to fix cmake issue
Signed-off-by: Gu, Jinghui <jinghui.gu@intel.com>
* Fix cmake issue caused by USE_MKL option
Signed-off-by: Gu, Jinghui <jinghui.gu@intel.com>
* Correct comments in MKL cmake file
Signed-off-by: Gu, Jinghui <jinghui.gu@intel.com>
2018-04-22 21:58:14 -07:00
