Commit Graph

96 Commits (516317cf91037698d53732769d3dd3431760ab3a)

Author SHA1 Message Date
tensor-tang ad45a08351
fix avx option (#16683)
6 years ago
Kaipeng Deng 54474637ae
Merge pull request #16057 from heavengate/softmax_axis
6 years ago
dengkaipeng 3e352388eb fix format. test=develop
6 years ago
dengkaipeng eb2123e12d fix doc and jit. test=develop
6 years ago
tensor-tang 1eff834e97
update jitkernel doc (#16327)
6 years ago
dengkaipeng 90bd038d35 fix format. test=develop
6 years ago
dengkaipeng f45aced59b add jit test. develop=test
6 years ago
dengkaipeng 51536f7f52 StrideASum. test=develop
6 years ago
dengkaipeng 93701dba50 add jit kernel for softmax axis. test=develop
6 years ago
tensor-tang 14d871121b enhance jitkernel unit test
6 years ago
tensor-tang cfc83c1445 refine jitcodekey and enhance unit tests
6 years ago
tensor-tang 45bdd84dac enhance the jitkernel helper and add unit tests
6 years ago
tensor-tang 14a764c930 simplify the jitkernel templates and tests
6 years ago
tensor-tang 802f362ac4 unify the kernelfuncs cache and add unit test
6 years ago
tensor-tang b16dabd7e0 refine vbroadcast jitcode
6 years ago
tensor-tang 641b3cccce add vbroadcast mkl code and jitcode
6 years ago
tensor-tang 41a1270856 add vbroadcast jitkernel refer code and use it
6 years ago
tensor-tang 867e93b21a add jitkernel vcopy and speedup unit test time
6 years ago
tensor-tang 8bc6381546 fix jitcodekey and refine test
6 years ago
tensor-tang 7044cfa7c7 add sgd jitcode and op test
6 years ago
tensor-tang 8e04133719 add benchmark and mkl sgd implement
6 years ago
tensor-tang a0c37662b9 enable sgd jitkernel refer code and test
6 years ago
Yiqun Liu 7d96c74ab2
Initialize the benchmark tester for operator. (#15772)
6 years ago
tensor-tang 5aea2cd2e0
Merge pull request #15652 from tensor-tang/refine/pyramiddnn
6 years ago
Yihua Xu 685a20ef56 Add JIT CRF_decoding and Layer_norm unit-test (#15699)
6 years ago
tensor-tang 75fc792d40 fix when table width larger than 64
6 years ago
tensor-tang 40402d5e68 add emb seqpool jitcode
6 years ago
tensor-tang a3a3d3d861 add embseqpool jitkernel mkl impl and use it
6 years ago
tensor-tang 15da2f9a0d add embseqpool jitkernel refer code, test and benchmark
6 years ago
tensor-tang 33d0cebbff
Merge pull request #15695 from tensor-tang/fix/name
6 years ago
tensor-tang fb2a7b2300
fix aligned-new error in jitkernel (#15626)
6 years ago
tensor-tang 15d7220f94 fix jitcode name
6 years ago
Gabor Buella da9c94da33 Clang build fixes (#15628)
6 years ago
tensor-tang a6a1a92ef7
Merge pull request #15586 from tensor-tang/jit/cache
6 years ago
tensor-tang ba02ac4692
use mat attr and refine test (#15448)
6 years ago
tensor-tang 2b0811c3fb refine vadd jitkernel choice
6 years ago
tensor-tang 6e1ee7fb57 cache softmax kernel func
6 years ago
tensor-tang d59f733551 refine softmax and use with cache
6 years ago
tensor-tang 7383eefd2d add softmax mix and mkl code
6 years ago
tensor-tang 50945685f2 add hmax, hsum jitcode
6 years ago
tensor-tang 8117725852 add jit kernel hsum, hmax and softmax refer code
6 years ago
tensor-tang b67584a6e9 jit benchmark use tensor
6 years ago
tensor-tang 316e44b1b7 fix unused warnings
6 years ago
tensor-tang 579d758254 fix jitkernel tests and refine benchmark
6 years ago
tensor-tang 93e75c5ae5 refine jitcode of vsub and vsquare
6 years ago
tensor-tang 09c5786e22 add square jitkernel
6 years ago
tensor-tang 266a5d2f52 implement matmul refer and mkl kernel
6 years ago
tensor-tang c5623c87a3 init jit matmul kernel
6 years ago
tensor-tang c3a9f3c4b2 fix typo and refine
6 years ago
tensor-tang 123b98f417 refine heigth and codesize and support all pool
6 years ago