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