Megvii Engine Team
1783b8977a
feat(profiler): integrate cupti backend
GitOrigin-RevId: dec8be1908
3 years ago
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
597a1e791b
refactor(imperative): add interface to clear algorithm cache
GitOrigin-RevId: 662618954b
3 years ago
Megvii Engine Team
dbce6526d6
fix(mge/functional): fix return dtype of comparison function
GitOrigin-RevId: 810e32a829
3 years ago
Megvii Engine Team
b92866d2c2
fix(build): fix build depends dirty file issue
GitOrigin-RevId: 435d8b5c50
3 years ago
Megvii Engine Team
27d4c4b36c
refactor(stats): use static inline variable declaration
GitOrigin-RevId: 7d86e5f257
3 years ago
Megvii Engine Team
787a22a9d6
perf(tensor): implement __new__ in cpp
GitOrigin-RevId: 4defd249c3
3 years ago
Megvii Engine Team
53d48ca399
fix(fastrun): persistent cache add prefix
(cherry picked from commit bdd82c1976
)
GitOrigin-RevId: 0b13d32175
3 years ago
Megvii Engine Team
7005b7d691
fix(mge): fix fastpath check
GitOrigin-RevId: acc8957469
3 years ago
Megvii Engine Team
7bf5b0ee1e
test(imperative): check env values after each pytest
GitOrigin-RevId: 826788113a
3 years ago
Megvii Engine Team
811579b0e7
fix(imperative): restrict value converts to symbolvar
GitOrigin-RevId: 271267be69
3 years ago
Megvii Engine Team
99a85c4079
fix(mge): fix advanced indexing grad
GitOrigin-RevId: 8033c9322d
3 years ago
Megvii Engine Team
409c988163
fix(imperative): add matmul apply_on_varnode
GitOrigin-RevId: 2cf6bf237c
3 years ago
Megvii Engine Team
7927e98fd6
perf(mge): speed up PixelShuffle
GitOrigin-RevId: 942e755745
3 years ago
Megvii Engine Team
43e5f41c4b
docs(mgb/docs): add alternative interface for deprecated api
GitOrigin-RevId: fdeaa5b64f
3 years ago
Megvii Engine Team
f7e10ea8d7
perf(imperative): improve matmul/batch_matmul
GitOrigin-RevId: 4ceb2eb601
3 years ago
Megvii Engine Team
1c2a323e78
feat(mge): add warning message when mismatched cuda sm is detected
GitOrigin-RevId: f78c79eb06
3 years ago
Megvii Engine Team
877bda4180
perf(mge): improve cross stream memory borrowing
GitOrigin-RevId: c68977c5dc
3 years ago
Megvii Engine Team
84ce94fbd9
docs(imperative): fix docs about vision related function
GitOrigin-RevId: 55d12d5e7a
3 years ago
Megvii Engine Team
c2435d1561
perf(imperative): specialize adaptive pooling
GitOrigin-RevId: 01e1418458
3 years ago
Megvii Engine Team
8fcbe8258d
docs(mge/functional): fix functional.eye docstring
GitOrigin-RevId: 2645af5e35
3 years ago
Megvii Engine Team
d2278f02d2
perf(imperative): speed up conv_transpose3d
GitOrigin-RevId: e741305446
3 years ago
Megvii Engine Team
3f11b4213a
docs(functional): replace megengine function testcode with doctest format
GitOrigin-RevId: f91d5f5afd
3 years ago
Megvii Engine Team
60163c5d12
docs(functional): replace elem testcode with doctest format
GitOrigin-RevId: a32a011724
3 years ago
Megvii Engine Team
fd5031f8eb
docs(functional): replace utils testcode with doctest format
GitOrigin-RevId: 8618918ff3
3 years ago
Megvii Engine Team
aa63230592
docs(functional): replace loss function testcode with doctest format
GitOrigin-RevId: 98224f0e5f
3 years ago
Megvii Engine Team
e8d0f9dbf2
docs(example): replace dist.help testcode with doctest format
GitOrigin-RevId: 79a7b72ca3
3 years ago
Megvii Engine Team
f5f9249a16
docs(compose): update compose API docstring and example
GitOrigin-RevId: fd52df2c48
3 years ago
Megvii Engine Team
7d3a6db0ef
docs(tensor): add more introduction about Tensor
GitOrigin-RevId: 4eba1e6c03
3 years ago
Megvii Engine Team
90cedd3c1d
fix(imperative): restrict using convert_inputs in py_apply
GitOrigin-RevId: b021aac8a6
3 years ago
Megvii Engine Team
2d72de8a87
fix(mge): fix infer output attrs fallible
GitOrigin-RevId: ea18c7f753
3 years ago
Megvii Engine Team
fd6f8e58b0
feat(mgb/dtype): add dtype qint1
GitOrigin-RevId: abe9fb68b1
3 years ago
Megvii Engine Team
f7a5fe17d3
feat(mge/config): add a config option for memory forwarding
GitOrigin-RevId: 657326154c
3 years ago
Megvii Engine Team
69673f14f7
fix(imperative): remove convert_inputs from concat
GitOrigin-RevId: 1511cb4b43
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
2a900a69cb
perf(imperative): improve reduce op performance
GitOrigin-RevId: 26d982a7b8
3 years ago
Megvii Engine Team
22be7e11ca
docs(mge/functional): typo
GitOrigin-RevId: 0ebc6cb322
3 years ago
Megvii Engine Team
111c150db1
docs(mge/functional): fix typo
GitOrigin-RevId: 9473f17970
3 years ago
Megvii Engine Team
4f60fbbb2f
fix(mge/distributed): add polling to solve xmlrpc server io error
GitOrigin-RevId: ca19b1566b
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
bc9aa47ad8
feat(mge/indexing): support newaxis
GitOrigin-RevId: 8338c4b475
3 years ago
Megvii Engine Team
9779bc7f6d
fix(imperative): allow rng op infer shape fallible
GitOrigin-RevId: 687844500c
3 years ago
Megvii Engine Team
6c413ba943
refactor(mge): refactor physical tensor
GitOrigin-RevId: 93ba67ca5a
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
e64536a31e
fix(imperative): fix the dtype promote problem when amp
GitOrigin-RevId: 43e1035fc8
3 years ago
Megvii Engine Team
2f3bc2db9d
perf(mge/utils): move astensor1d into C++
GitOrigin-RevId: e7c6659020
3 years ago
Megvii Engine Team
fa62f6c06e
perf(mge/utils): move convert_input into C++
GitOrigin-RevId: 0d1cd36251
3 years ago
Megvii Engine Team
d98be08030
perf(mge): move Const into C++
GitOrigin-RevId: 31a443cffd
3 years ago
Megvii Engine Team
1709b3940b
perf(mge/functional): speed up Broadcast and Reshape
GitOrigin-RevId: a72f5460b6
3 years ago