Megvii Engine Team
fd6f8e58b0
feat(mgb/dtype): add dtype qint1
GitOrigin-RevId: abe9fb68b1
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
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
Megvii Engine Team
0f736a0ab4
perf(mge/functional): speed up Dimshuffle
GitOrigin-RevId: 8160c9522b
3 years ago
Megvii Engine Team
3e5e08b0b4
perf(mge/functional): speed up RemoveAxis
GitOrigin-RevId: 9c5d27fe1d
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
87f00232f2
fix(mge/gm): fix missing dtype checking while attach tensors
GitOrigin-RevId: f0aaea99b9
3 years ago
Megvii Engine Team
ca00177719
perf(dispatch): speed up dispatch system
GitOrigin-RevId: eabbe3e021
3 years ago
Megvii Engine Team
59cbf9583d
fix(subgraph): use CompiledOp in cpu to avoid workspace error
GitOrigin-RevId: 104dd982ef
3 years ago
Megvii Engine Team
b6ce02a152
fix(subgraph): fallback back to cg if jit unsupported
GitOrigin-RevId: 853a00a402
3 years ago
Megvii Engine Team
21f5a7fcc0
fix(subgraph): fix device recognition and scalar propagate
GitOrigin-RevId: fd2fe8bec9
3 years ago
Megvii Engine Team
2775f4580c
feat(subgraph): subgraph builder supports jit and custom grad
GitOrigin-RevId: e1a1ebdf1c
3 years ago
Megvii Engine Team
ca2deebc0f
fix(imperative/tensor): make @ operator has the same functionality as matmul functional
GitOrigin-RevId: bf6136cc1a
3 years ago
Megvii Engine Team
e860a08386
refactor(mge/indexing): move indexing into c++
GitOrigin-RevId: 43fbdb22dd
3 years ago
Megvii Engine Team
c34a75d0f4
fix(trace): assume result is not scalar when shape is valid
GitOrigin-RevId: beee2d0f28
3 years ago
Megvii Engine Team
68cde8734e
fix(mge/imperative): support broadcast with None
GitOrigin-RevId: dd330a2a1d
3 years ago
Megvii Engine Team
0bdd0b1467
refactor(dispatch): switch to new dispatch system
GitOrigin-RevId: 32dd49a23a
3 years ago
Megvii Engine Team
b75d10094d
fix(mge/elwise): removed back to fp32 mode
GitOrigin-RevId: 6d1720f4af
3 years ago
Megvii Engine Team
9fd2e66350
feat(mge/elwise): removed back to fp32 mode
GitOrigin-RevId: a665a279a6
3 years ago
Megvii Engine Team
8a222d2c9c
fix(mge): replace functional paramter
GitOrigin-RevId: af3c68c588
3 years ago
Megvii Engine Team
a404c508e9
feat(mge): support dump with specific format
GitOrigin-RevId: 57a7c0de02
3 years ago
Megvii Engine Team
cebda6ff8d
feat(mge/imperative): add ctc loss
GitOrigin-RevId: e29854a98e
3 years ago
Megvii Engine Team
27f2ecc436
feat(imperative/opr): add extern c opr and support dump
GitOrigin-RevId: b29e374c60
3 years ago
Megvii Engine Team
c5f8b58385
docs(docstring): transfer to google style
GitOrigin-RevId: a71245c553
3 years ago
Megvii Engine Team
5431929ec7
feat(functional): let advance indexing support empty tensor and add more tests
GitOrigin-RevId: 49e1492934
3 years ago
Megvii Engine Team
a7a3bf2d6c
test(subgraph): simple test for subgraph
GitOrigin-RevId: 3d6ecd5db7
3 years ago
Megvii Engine Team
d063d5774f
perf(functional): use fma to reduce elemwise but disable subgraph compilation
GitOrigin-RevId: c75a6e1a09
3 years ago
Megvii Engine Team
2a063f8e87
fix(subgraph): fix scope mismatch of subgraph content
GitOrigin-RevId: 6e23456250
3 years ago
Megvii Engine Team
8c47c1f149
perf(syncbn): reimplement with subgraph
GitOrigin-RevId: 13e7e3d3c0
3 years ago
Megvii Engine Team
bfc4e7a966
docs(mge): fix amp docstring problems
GitOrigin-RevId: e5540eb940
3 years ago
Megvii Engine Team
070c811732
fix(imperative): remove convert_inputs
GitOrigin-RevId: a3c43db746
4 years ago
Megvii Engine Team
1bf18252c4
feat(mge/amp): add mix precision autocast support
GitOrigin-RevId: 6fbffc4845
4 years ago
Megvii Engine Team
dedecf6922
fix(imperative/utils): fix logical error of replace var
GitOrigin-RevId: 614302552c
4 years ago
Megvii Engine Team
ffe2bb2eb2
fix(mge): fix some errors caused by unknown shape when using symbolic trace or building graph
GitOrigin-RevId: 70ddc06eee
4 years ago
Megvii Engine Team
2cd9823210
fix(mgb/tensorrt): fix trt runtime, padding channel to a multiple of 4 when using kCHW4 IOFormat
GitOrigin-RevId: c5f1ed70da
3 years ago
Megvii Engine Team
54a4d70eb5
feat(src/serialization): add support of serializing metadata
GitOrigin-RevId: b563c94451
4 years ago
Megvii Engine Team
241b35a697
refactor(ops): remove BackwardGraph op
GitOrigin-RevId: eda20e5760
4 years ago
Megvii Engine Team
8be70d170c
docs(misc): correct docstring format broadly
GitOrigin-RevId: 45234ca07e
4 years ago
Megvii Engine Team
984d85ca26
feat(mge/functional): argmin and argmax support negtive axis
GitOrigin-RevId: a1bd1102a6
4 years ago
Megvii Engine Team
87d6ff228d
fix(mge): fix dumping backward graph
GitOrigin-RevId: 430f110053
4 years ago
Megvii Engine Team
6972896933
fix(mge/utils): fix network multiple outputs issue
GitOrigin-RevId: d22e639cd3
4 years ago
Megvii Engine Team
13e6ea349d
feat(imperative/opr): rebase rng refactoring to dev & add python module
GitOrigin-RevId: ee5984c52d
4 years ago
Megvii Engine Team
65af9cfcdc
refactor(mge): use lower case for default string parameters in functional and module
GitOrigin-RevId: dbc1f27ff7
4 years ago
Megvii Engine Team
9dbe71dd1a
feat(mge/utils): add array method for varnode
GitOrigin-RevId: 6e4d05b475
4 years ago
Megvii Engine Team
5a7c30e0f7
feat(mge/tools): add summary print for module_stats and network_visualize
GitOrigin-RevId: 7d85aa0ea2
4 years ago
Megvii Engine Team
702ed9eee6
feat(mge): enable cg dump for debugging
GitOrigin-RevId: 4292364d0d
4 years ago