Megvii Engine Team
c2293815b2
fix(autodiff): proxy_graph_detail::make_backward_graph support multiple opnodes
GitOrigin-RevId: 2c0c8f330d
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
6c413ba943
refactor(mge): refactor physical tensor
GitOrigin-RevId: 93ba67ca5a
3 years ago
Megvii Engine Team
d56570d929
fix(megbrain): add rdnn to copybara
GitOrigin-RevId: 7d8bf77053
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
a4327c4d25
perf(imperative): add dim_expansion transform for conv/bn1d
GitOrigin-RevId: d14a69424d
3 years ago
Megvii Engine Team
72a70dd6a7
perf(imperative): specialize convolution implementation
GitOrigin-RevId: 33634c550f
3 years ago
Megvii Engine Team
0a6f4a880e
fix(mge/dtr): fix dtr problem
GitOrigin-RevId: 2a703f9ee4
3 years ago
Megvii Engine Team
529b394f9c
fix(imperative): fix profiler problem
GitOrigin-RevId: ea0d8a5054
3 years ago
Megvii Engine Team
e64536a31e
fix(imperative): fix the dtype promote problem when amp
GitOrigin-RevId: 43e1035fc8
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
730ddc2d81
perf(interpreter): improve interpreter performance
GitOrigin-RevId: 88f51d15f8
3 years ago
Megvii Engine Team
729242f9f8
refactor(imperative): move typecvt code of sereval ops to c++
GitOrigin-RevId: 4ffaa376c1
3 years ago
Megvii Engine Team
3c3fc6f33c
refactor(imperative): move python code of elemwise/reduce/conv2d/bn to c++
GitOrigin-RevId: 01b5324392
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
8c2b916ef5
refactor(imperative): remove some methods in proxy graph
GitOrigin-RevId: 1fb68a1da2
4 years ago
Megvii Engine Team
2348a963f2
refactor(imperative): apply workspace limit hook to mini graph
GitOrigin-RevId: 27c51f3147
4 years ago
Megvii Engine Team
fea46ea9a4
perf(imperative): add opr cache for apply_on_physical_tensor
GitOrigin-RevId: fc5d5fb34d
4 years ago
Megvii Engine Team
177001d5e5
refactor(dispatch): allow dynamic type creation
GitOrigin-RevId: 27dde05cff
4 years ago
Megvii Engine Team
150a6a6151
perf(dispatch/trace): remove unnecessary h2d for constant
GitOrigin-RevId: d00de3fc1f
4 years ago
Megvii Engine Team
81d8c73a41
perf(dispatch/trace): serval tricks to speed up trace
GitOrigin-RevId: 2bdd70cde2
4 years ago
Megvii Engine Team
4fa6162027
perf(dispatch): improve performance of dispatch system
GitOrigin-RevId: 860028e1af
4 years ago
Megvii Engine Team
ca00177719
perf(dispatch): speed up dispatch system
GitOrigin-RevId: eabbe3e021
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
62034fb262
fix(imperative): make CompNode finalize happens before global object destructor
GitOrigin-RevId: 9a1f507c69
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
2775f4580c
feat(subgraph): subgraph builder supports jit and custom grad
GitOrigin-RevId: e1a1ebdf1c
4 years ago
Megvii Engine Team
3c61e0e02a
feat(ops): add JITFusion op
GitOrigin-RevId: 7dc35d4e80
4 years ago
Megvii Engine Team
aa587446fc
feat(subgraph): support shape inference for CompiledOp
GitOrigin-RevId: a96b8f3446
4 years ago
Megvii Engine Team
1c1e9b002d
fix(rng): init layout strides
GitOrigin-RevId: 9833d866da
4 years ago
Megvii Engine Team
9527859cc8
feat(opcache): add ndim and has_value to cache key
GitOrigin-RevId: ad073d389e
4 years ago
Megvii Engine Team
cbb47089a6
perf(interpreter): add fastpath for GetVarShape
GitOrigin-RevId: d1ac4e7fe3
4 years ago
Megvii Engine Team
b458178847
feat(opr): add mutable tensor opr
GitOrigin-RevId: 7f8a3d7b66
4 years ago
Megvii Engine Team
93ceb80ad2
refactor(imperative): fix broadcast,reshape,reduce
GitOrigin-RevId: ee3dc1487d
4 years ago
Megvii Engine Team
e6706be23a
refactor(imperative): remove infer_output_mem_desc
GitOrigin-RevId: bff62b33a0
4 years ago
Megvii Engine Team
a5af35c18c
refactor(imperative): remove command buffer
GitOrigin-RevId: 83c8cb6d3b
4 years ago
Megvii Engine Team
c34a75d0f4
fix(trace): assume result is not scalar when shape is valid
GitOrigin-RevId: beee2d0f28
4 years ago
Megvii Engine Team
95ac055538
feat(dnn,mgb,imperative): add diag opr implement
GitOrigin-RevId: 43016ffa2b
4 years ago
Megvii Engine Team
cbbca5fb10
feat(mge): add softmax op use cudnn api
GitOrigin-RevId: 7734ebf8c4
4 years ago