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
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
