Megvii Engine Team
1192a9a61b
fix(imperative): fix adaptive pool2d
GitOrigin-RevId: 624b47a51a
3 years ago
Megvii Engine Team
24c5c19bf0
fix(imperative): make functional ops support negative axis
GitOrigin-RevId: f61e01270b
3 years ago
Megvii Engine Team
f96429c031
feat(imperative): support empty tensor in roi_align
GitOrigin-RevId: aeb2770401
3 years ago
Megvii Engine Team
df5ebd3da7
fix(imperative/ops): fix the vmemory problem in 1.9
GitOrigin-RevId: 728e983701
3 years ago
Megvii Engine Team
8563f51404
fix(imperative): fix buildin reduce keepdim
GitOrigin-RevId: 38d90ab38a
3 years ago
Megvii Engine Team
da91e650a5
refactor(ops/layer_norm): speed up the host speed of layer_norm
GitOrigin-RevId: 6f359b5b29
3 years ago
Megvii Engine Team
15c6da6218
feat(imperative/amp): add nhwc support for adaptive pooling
GitOrigin-RevId: 7c5755308e
3 years ago
Megvii Engine Team
4aa79c453b
perf(mge): override grad of matmul
GitOrigin-RevId: d9d97e70fe
3 years ago
Megvii Engine Team
98b5ee78c1
feat(mge/dnn): add lamb optimizer
GitOrigin-RevId: 5a27157456
3 years ago
Megvii Engine Team
c2e9860feb
chore(license): remove all license in file header
GitOrigin-RevId: a0e31247a6
3 years ago
Megvii Engine Team
8182af6eb6
fix(mgb): fix strategy of grad_op and opr_attr
GitOrigin-RevId: bb7ab8fa9d
3 years ago
Megvii Engine Team
4b27e861f4
fix(ops): implement from_op_node for reshape
GitOrigin-RevId: 4c99438504
3 years ago
Megvii Engine Team
27d4c4b36c
refactor(stats): use static inline variable declaration
GitOrigin-RevId: 7d86e5f257
3 years ago
Megvii Engine Team
7bf5b0ee1e
test(imperative): check env values after each pytest
GitOrigin-RevId: 826788113a
3 years ago
Megvii Engine Team
409c988163
fix(imperative): add matmul apply_on_varnode
GitOrigin-RevId: 2cf6bf237c
3 years ago
Megvii Engine Team
275f12c98a
fix(mge): fix dimshuffle shape infer
GitOrigin-RevId: ad7a73fbd1
3 years ago
Megvii Engine Team
e59b6e13a3
fix(imperative/src): fix empty_tensor bug of convbwd&rng
GitOrigin-RevId: 4c948f41f0
3 years ago
Megvii Engine Team
7927e98fd6
perf(mge): speed up PixelShuffle
GitOrigin-RevId: 942e755745
3 years ago
Megvii Engine Team
f7e10ea8d7
perf(imperative): improve matmul/batch_matmul
GitOrigin-RevId: 4ceb2eb601
3 years ago
Megvii Engine Team
877bda4180
perf(mge): improve cross stream memory borrowing
GitOrigin-RevId: c68977c5dc
4 years ago
Megvii Engine Team
c2eec47b42
fix(imperative/src): fix adaptive_pooling bug
GitOrigin-RevId: e1dc46e3fa
3 years ago
Megvii Engine Team
c2435d1561
perf(imperative): specialize adaptive pooling
GitOrigin-RevId: 01e1418458
3 years ago
Megvii Engine Team
d2278f02d2
perf(imperative): speed up conv_transpose3d
GitOrigin-RevId: e741305446
3 years ago
Megvii Engine Team
3a5347ed21
perf(imperative): speed up pooling
GitOrigin-RevId: 9f60b45eeb
3 years ago
Megvii Engine Team
d9c4ef59fe
perf(imperative): using simple hash key in heuristic cache
GitOrigin-RevId: 6fddd612e7
3 years ago
Megvii Engine Team
26ea33c6a7
perf(imperative): improve convbwd performance
GitOrigin-RevId: cfc8623d7a
3 years ago
Megvii Engine Team
2d72de8a87
fix(mge): fix infer output attrs fallible
GitOrigin-RevId: ea18c7f753
3 years ago
Megvii Engine Team
4d5faa3f0a
fix(imperative): using DnnOprCaller to avoid early destruction of dnn_opr
GitOrigin-RevId: 4a14f53738
3 years ago
Megvii Engine Team
da620ca193
perf(imperative): specialize batchnorm implementation
GitOrigin-RevId: 83a8259044
3 years ago
Megvii Engine Team
2a900a69cb
perf(imperative): improve reduce op performance
GitOrigin-RevId: 26d982a7b8
3 years ago
Megvii Engine Team
273c0e8745
fix(autodiff): fix some bugs in relation to 2nd order grad
1. implement double backward for batchnorm
2. fix grad attach in nested grad manager
3. pad empty tensor for unsatisfied output_has_grad
4. support double backward for jit subgraph
5. support double backward for autodiff.Function
6. readd debug flag MGE_LOG_OP_DISPATCH
GitOrigin-RevId: cd31ddc620
3 years ago
Megvii Engine Team
9779bc7f6d
fix(imperative): allow rng op infer shape fallible
GitOrigin-RevId: 687844500c
3 years ago
Megvii Engine Team
8fa9a8defa
fix(imperative): fix dot-op implement
GitOrigin-RevId: b97290e1fc
3 years ago
Megvii Engine Team
7de1bb11ab
fix(mge/utils): disable memory forwarding for subgraph
GitOrigin-RevId: 3be6d5fe6b
3 years ago
Megvii Engine Team
b7c9361f81
perf(mge/functional): add infer_output_attrs_fallible for some ops
GitOrigin-RevId: 33ae4b18e9
3 years ago
Megvii Engine Team
72a70dd6a7
perf(imperative): specialize convolution implementation
GitOrigin-RevId: 33634c550f
3 years ago
Megvii Engine Team
2b80806f21
perf(imperative/src): improve dot performance
GitOrigin-RevId: 35b5bd164f
3 years ago
Megvii Engine Team
1709b3940b
perf(mge/functional): speed up Broadcast and Reshape
GitOrigin-RevId: a72f5460b6
3 years ago
Megvii Engine Team
3e206d899b
perf(mge/functional): speed up Split
GitOrigin-RevId: 43550a0706
3 years ago
Megvii Engine Team
8446626193
perf(imperative/src): improve elemwise
GitOrigin-RevId: 78aa487277
3 years ago
Megvii Engine Team
e400b7ffe5
perf(imperative): enable memory forwarding for imperative
GitOrigin-RevId: 7c1993979c
4 years ago
Megvii Engine Team
84d1a440f0
fix(imperative): do not use output_desc in rng ops
GitOrigin-RevId: e6a399be17
3 years ago
Megvii Engine Team
0cb60d646d
feat(imperative): add output_descs for apply_on_physical_tensor
GitOrigin-RevId: 5b036c2c5a
3 years ago
Megvii Engine Team
c7ded2fe2f
refactor(imperative): remove unnecessary reverve in small vector
GitOrigin-RevId: 85c30bc828
3 years ago
Megvii Engine Team
177001d5e5
refactor(dispatch): allow dynamic type creation
GitOrigin-RevId: 27dde05cff
4 years ago
Megvii Engine Team
7bd848ce04
fix(subgraph): fix hand-written backward for serval jit-elemwise ops
GitOrigin-RevId: ea3a40d96e
4 years ago
Megvii Engine Team
7be7656c9f
fix(imperative): explicitly manage global structures
GitOrigin-RevId: 0f910c34b6
4 years ago
Megvii Engine Team
b6ce02a152
fix(subgraph): fallback back to cg if jit unsupported
GitOrigin-RevId: 853a00a402
4 years ago
Megvii Engine Team
225045236b
perf(imperative): improve shape inference
GitOrigin-RevId: 98b4d7e9af
4 years ago
Megvii Engine Team
3c61e0e02a
feat(ops): add JITFusion op
GitOrigin-RevId: 7dc35d4e80
4 years ago