Commit Graph

120 Commits (cd327e66df21ffdbc9495369d6e1799551ea8e70)

Author SHA1 Message Date
Jacek Czaja c2efdfd5bc [MKL-DNN] Extending reusing to Elementwise_add_mkldnn op (#18146)
6 years ago
Yiqun Liu 660c1a65f3
Optimize fused_elewise_activation_grad op. (#18041)
6 years ago
Tao Luo 53fd507bae
fix merge conflict of 'Remove attribute in Allocator::Allocate' and elementwise_add_mkldnn_op (#17949)
6 years ago
Yihua Xu 9b5017366a Fix the format issue when 'X' is not nchw. (#17833)
6 years ago
lvmengsi 10b23a72c1 Double backward elementwise div (#17416)
7 years ago
lvmengsi 977e9fcb27
support elementwise_sub double backward (#17476)
7 years ago
Kaipeng Deng bd9bef5a4e
add elementwise_add_grad_grad op (#17366)
7 years ago
Yiqun Liu dcda20233c
Optimize the elementwise op using eigen (#15494)
7 years ago
Kaipeng Deng 8bae8590ac
add double grad for elementwise_mul op (#17255)
7 years ago
zhoukunsheng 4292bd8687 Mod floordiv (#17251)
7 years ago
zhaoyuchen2018 792443ef23
Refine elementwise kernel. (#16952)
7 years ago
gongweibao 91784f8ec3
Fix code in document. (#17237)
7 years ago
Zeng Jinle ee2028a110
Add use_cuda to inplace pass (#17205)
7 years ago
Zeng Jinle 1c526e1d1a
Fix some grad op desc makers (#16633)
7 years ago
Jacek Czaja 2632327429 [MKL-DNN] Tensor modifications revert (#16462)
7 years ago
Zeng Jinle c7c6eeb44e
Merge pull request #16409 from sneaxiy/feature/advance_gc
7 years ago
liuwei1031 8d22bc17a4
Memory optimize (#16410)
7 years ago
sneaxiy 072d95d8f6 Merge develop
7 years ago
sneaxiy a93a9eef8f add op registry type
7 years ago
phlrain 5dc9b51994 fix time; test=develop
7 years ago
phlrain 18d107c27a add floordiv and mod op; test=develop
7 years ago
phlrain 56c2d384c7 add elementwise floordiv, mod; test=develop
7 years ago
tensor-tang 14a764c930 simplify the jitkernel templates and tests
7 years ago
tensor-tang 802f362ac4 unify the kernelfuncs cache and add unit test
7 years ago
Jacek Czaja c63f6b2039 - MKL-DNN pooling updated to set_prim_desc
7 years ago
dzhwinter 104d3b4e68 add details. test=develop
7 years ago
dzhwinter 94dd50c33f add details. test=develop
7 years ago
dzhwinter 9c9ad7d40b Merge remote-tracking branch 'origin/develop' into feature/ir_inplace_pass
7 years ago
Krzysztof Binias 69b7c595d6 Small fix
7 years ago
Krzysztof Binias b1bdcd4de8 Make separate folders for mkldnn codes
7 years ago
dzhwinter ee3aae56cd merge develop branch. test=develop
7 years ago
chengduo bf91d11ed5
Clean elementwise_op_function (#15502)
7 years ago
dzhwinter 8f3b252392 squash commits. test=develop
7 years ago
Xin Pan 3ecf6bb338
Merge pull request #15028 from yihuaxu/develop_641313ea7_elementwise_mul_mkldnn_bug_fix
7 years ago
Wu Yi fd85418329
[Feature] support mix precision training for resnet (#14899)
7 years ago
sneaxiy 73896eeb94 merge develop
7 years ago
sneaxiy b56aca82e9 merge develop
7 years ago
sneaxiy dde3afe7b7 Merge develop
7 years ago
Wu Yi 856f0da0fe
Fp16 training (#14992)
7 years ago
Yihua Xu d4606bcb22 Fix the exception when tensor format is x
7 years ago
peizhilin 2e35290ffa fix build issue
7 years ago
peizhilin 201283f95c fix code style
7 years ago
peizhilin 9e60c58666 Merge remote-tracking branch 'upstream/develop' into windows/mkl
7 years ago
tensor-tang 693e5e65ce
Merge pull request #14958 from tensor-tang/refine/jit
7 years ago
sneaxiy c1f7e54f62 merge develop
7 years ago
typhoonzero da87f7a698 Revert "[Feature] Fp16 training for resnet50 (#14850)"
7 years ago
tensor-tang 1aaec571c2 fix enum style
7 years ago
tensor-tang f5532877f7 Merge remote-tracking branch 'ups/develop' into refine/jit
7 years ago
Wu Yi 3d750f9c5a
[Feature] Fp16 training for resnet50 (#14850)
7 years ago
sneaxiy 045dc12766 merge develop
7 years ago
peizhilin 9f55f1ff50 use the platform api to decide the specific instruction support or not
7 years ago
sneaxiy ae6f46a1a9 rewrite variable type
7 years ago
peizhilin 0b4f742e8a fix the build issue
7 years ago
peizhilin da42cf2055 fix build issue when xbyak is disabled on windows
7 years ago
peizhilin 1cc9d59838 disable xbyak on windows
7 years ago
tensor-tang 893957f711 Merge remote-tracking branch 'ups/develop' into refine/jit
7 years ago
tensor-tang 6648995f53 fix build
7 years ago
sneaxiy a500dfa579 rewrite ddim
7 years ago
peizhilin fa135bbf52 Fix the mkl build script on windows
7 years ago
peizhilin b601f2de8d include the mkl fix only
7 years ago
peizhilin 5a6d7fe2ff add mkl,ctc support for windows
7 years ago
tensor-tang 74292f414c enable eltwise nchw16c mul nc
7 years ago
Yu Yang 9bd70a1e04 Change tensor uses proto::VarType::type
7 years ago
Yu Yang 7604b1ad51 Fix Eigen macro when using GPU
7 years ago
chengduo 04539d4c5d
Fix clip.py (#14718)
7 years ago
Michal Gallus 9455be0ba5 EltwiseMul: Extract StringToFormat to MKLDNN helper
7 years ago
Michal Gallus c69c41604e MKLDNN elementwise_mul: Move Kernel to KernelPool to avoid segfaults
7 years ago
Michal Gallus 08f63c4d12 MKLDNN elementwise_mul: Lint changes to UT & integration
7 years ago
Michal Gallus ed31936ba1 MKLDNN elementwise_mul: Support NCHW, update UT
7 years ago
Wu Yi a2d9b34417
Refine operator cmake (#14413)
7 years ago