Megvii Engine Team
58df717e37
fix(mge/autodiff): fix attaching tensor already in gradient path
GitOrigin-RevId: da774509ca
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
4fb3d886b8
perf(misc): use reinterpret_cast to convert valueshape
GitOrigin-RevId: ee14c7ce5e
3 years ago
Megvii Engine Team
27d4c4b36c
refactor(stats): use static inline variable declaration
GitOrigin-RevId: 7d86e5f257
3 years ago
Megvii Engine Team
877bda4180
perf(mge): improve cross stream memory borrowing
GitOrigin-RevId: c68977c5dc
4 years ago
Megvii Engine Team
95a30eb6f6
perf(imperative): speed up stackmanager guard
GitOrigin-RevId: 12d23b6f7e
3 years ago
Megvii Engine Team
2a900a69cb
perf(imperative): improve reduce op performance
GitOrigin-RevId: 26d982a7b8
3 years ago
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
a4327c4d25
perf(imperative): add dim_expansion transform for conv/bn1d
GitOrigin-RevId: d14a69424d
3 years ago
Megvii Engine Team
0a6f4a880e
fix(mge/dtr): fix dtr problem
GitOrigin-RevId: 2a703f9ee4
3 years ago
Megvii Engine Team
730ddc2d81
perf(interpreter): improve interpreter performance
GitOrigin-RevId: 88f51d15f8
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
0cb60d646d
feat(imperative): add output_descs for apply_on_physical_tensor
GitOrigin-RevId: 5b036c2c5a
3 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
7be7656c9f
fix(imperative): explicitly manage global structures
GitOrigin-RevId: 0f910c34b6
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
9527859cc8
feat(opcache): add ndim and has_value to cache key
GitOrigin-RevId: ad073d389e
4 years ago
Megvii Engine Team
b458178847
feat(opr): add mutable tensor opr
GitOrigin-RevId: 7f8a3d7b66
4 years ago
Megvii Engine Team
e6706be23a
refactor(imperative): remove infer_output_mem_desc
GitOrigin-RevId: bff62b33a0
4 years ago
Megvii Engine Team
9ce1f0f5d1
refactor(dispatch): implement grad
GitOrigin-RevId: d8367f9587
4 years ago
Megvii Engine Team
c609c031f1
refactor(dispatch): implement symbol
GitOrigin-RevId: c7bd86f5c1
4 years ago
Megvii Engine Team
e32929dfd2
refactor(dispatch): implement scalar
GitOrigin-RevId: b244c2ca1a
4 years ago
Megvii Engine Team
59084fa857
refactor(dispatch): implement lazy_eval
GitOrigin-RevId: 4e3f3a1c44
4 years ago
Megvii Engine Team
d2b67c2a88
refactor(dispatch): implement trace
GitOrigin-RevId: f8d3005732
4 years ago
Megvii Engine Team
39ac606b9c
refactor(dispatch): implement eval
GitOrigin-RevId: 32563e0a27
4 years ago
Megvii Engine Team
d8d5edb31f
refactor(opdef): overwrite shared_from_this to avoid const_cast
GitOrigin-RevId: 5322dbbb3f
4 years ago
Megvii Engine Team
2e42bc083d
feat(dispatch): implement new op dispatch system
GitOrigin-RevId: 355da6b814
4 years ago
Megvii Engine Team
2be6ceda03
feat(imperative/utils): add serval utils
GitOrigin-RevId: f401663ae3
4 years ago
Megvii Engine Team
e7c2ed1183
fix(imperative): add missing pragma once
GitOrigin-RevId: 7a9fff1171
4 years ago
Megvii Engine Team
1657b8e881
fix(fastrun): fix persistent_cache in redis
GitOrigin-RevId: ada5862b05
4 years ago
Megvii Engine Team
6b28a2f9ec
fix(copybara): re-open redis cache
GitOrigin-RevId: 055bf6aa4f
4 years ago
Megvii Engine Team
60c14b683f
fix(copybara): remove redis client in imperative
GitOrigin-RevId: 86fe93f4c9
4 years ago
Megvii Engine Team
cf1db2616e
fix(fastrun): replace py_redis with cpp_redis to avoid deadlock
GitOrigin-RevId: 9af7fa5c97
4 years ago
Megvii Engine Team
a2a09ef990
fix(imperative): release dtr related resources when disable dtr
GitOrigin-RevId: eacfded9de
4 years ago
Megvii Engine Team
2676fb7344
refactor(imperative): remove enable_defrag interface as it is enabled by default
GitOrigin-RevId: de9e7d7f16
4 years ago
Megvii Engine Team
d909950f64
refactor(imperative): remove swap in dtr
GitOrigin-RevId: 5c9e42f74a
4 years ago
Megvii Engine Team
369c2ccc5a
style(all): reformat c++ code
GitOrigin-RevId: 3ffd1b211f
4 years ago
Megvii Engine Team
c674bf0e6e
feat(debug): add gdb-scripts
GitOrigin-RevId: 43f9674f50
4 years ago
Megvii Engine Team
d02b0a08f7
feat(lite): fix typo
GitOrigin-RevId: 8c46aa3a30
4 years ago
Megvii Engine Team
93f70a9539
feat(src): add MGB_CUSTOM_OP macro
GitOrigin-RevId: beddcf9745
4 years ago
Megvii Engine Team
90dd07161c
refactor(imperative): modify the python interface of custom op
GitOrigin-RevId: e82e5de480
4 years ago
Megvii Engine Team
cbf024bf20
feat(imperative): add adaptor between custom op and imperative runtime
GitOrigin-RevId: d7877f2e32
4 years ago