Megvii Engine Team
cca38c4e8d
fix(mge): fix fastpath check
GitOrigin-RevId: acc8957469
3 years ago
Megvii Engine Team
b9e850a479
test(imperative): check env values after each pytest
GitOrigin-RevId: 826788113a
3 years ago
Megvii Engine Team
d984be591d
fix(imperative): restrict value converts to symbolvar
GitOrigin-RevId: 271267be69
3 years ago
Megvii Engine Team
94960ecf42
fix(imperative): restrict using convert_inputs in py_apply
GitOrigin-RevId: b021aac8a6
3 years ago
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
bc9aa47ad8
feat(mge/indexing): support newaxis
GitOrigin-RevId: 8338c4b475
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
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
a4d473c99a
perf(mge/functional): speed up AddAxis
GitOrigin-RevId: 92a3e1bdd3
3 years ago
Megvii Engine Team
3e206d899b
perf(mge/functional): speed up Split
GitOrigin-RevId: 43550a0706
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
e400b7ffe5
perf(imperative): enable memory forwarding for imperative
GitOrigin-RevId: 7c1993979c
4 years ago
Megvii Engine Team
1ce78aa09b
fix(imperative): destruct dnn handles at last
GitOrigin-RevId: 7a67c68c55
3 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
3 years ago
Megvii Engine Team
150a6a6151
perf(dispatch/trace): remove unnecessary h2d for constant
GitOrigin-RevId: d00de3fc1f
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
b6ce02a152
fix(subgraph): fallback back to cg if jit unsupported
GitOrigin-RevId: 853a00a402
4 years ago
Megvii Engine Team
21f5a7fcc0
fix(subgraph): fix device recognition and scalar propagate
GitOrigin-RevId: fd2fe8bec9
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
e860a08386
refactor(mge/indexing): move indexing into c++
GitOrigin-RevId: 43fbdb22dd
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
cf5e9488bb
fix(traced_module): fix module trace transformation
GitOrigin-RevId: ce11fe5e09
4 years ago
Megvii Engine Team
de8ffe0c12
refactor(imperative): unify interpreter option setting
GitOrigin-RevId: 53510445cc
4 years ago
Megvii Engine Team
0bdd0b1467
refactor(dispatch): switch to new dispatch system
GitOrigin-RevId: 32dd49a23a
4 years ago
Megvii Engine Team
d3689c3f3c
feat(imperative/python): add transformation manager
GitOrigin-RevId: a3c1732ffd
4 years ago
Megvii Engine Team
619d78ed86
fix(imperative): check async error when getting value
GitOrigin-RevId: 52b8a29932
4 years ago
Megvii Engine Team
1999307015
feat(mgb/opr): add dropout kernel
GitOrigin-RevId: d248bd2005
4 years ago
Megvii Engine Team
1657b8e881
fix(fastrun): fix persistent_cache in redis
GitOrigin-RevId: ada5862b05
4 years ago
Megvii Engine Team
a72e0cb568
feat(imperative,src): add jit builder for custom op
GitOrigin-RevId: 3bb0b46311
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
3d6fe33e2d
Revert "fix(imperative): check async error when getting value"
This reverts commit 3945a9bfa2
GitOrigin-RevId: ec6d3574b2
4 years ago
Megvii Engine Team
2065f0c26c
fix(mge/subgraph): fix mem leak
GitOrigin-RevId: dc7bfe6265
4 years ago
Megvii Engine Team
57e197475b
fix(imperative): check async error when getting value
GitOrigin-RevId: 3945a9bfa2
4 years ago
Megvii Engine Team
f64a2e02c9
Revert "fix(mge/subgraph): fix mem leak"
This reverts commit 48beb625e1 .
GitOrigin-RevId: 5fc58d5903
4 years ago
Megvii Engine Team
ffbfe59c9f
fix(mge/subgraph): fix mem leak
GitOrigin-RevId: 48beb625e1
4 years ago