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 |
Megvii Engine Team
|
0a56a5b110
|
feat(cuda/comp_node): gets (maximum) GPU memory allocated/reserved
GitOrigin-RevId: da2cc22436
|
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
|
a404c508e9
|
feat(mge): support dump with specific format
GitOrigin-RevId: 57a7c0de02
|
4 years ago |
Megvii Engine Team
|
182ca25d45
|
feat(mge/imperative): add coalesce_all_free_memory mge api
GitOrigin-RevId: ea05b1e9a1
|
4 years ago |
Megvii Engine Team
|
676b205bd8
|
style(imperative): add bypass of clang-format for cg options
GitOrigin-RevId: 824d610631
|
4 years ago |
Megvii Engine Team
|
fe15239ac0
|
fix(imperative): fix error message for tensors with intbx data type
GitOrigin-RevId: cbb42f8127
|
4 years ago |
Megvii Engine Team
|
369c2ccc5a
|
style(all): reformat c++ code
GitOrigin-RevId: 3ffd1b211f
|
4 years ago |
Megvii Engine Team
|
000517c641
|
fix(grad): stop using exception in grad_override
GitOrigin-RevId: 00ae38d48b
|
4 years ago |
Megvii Engine Team
|
b6142bee9a
|
feat(imperative): support tensor with uint16 date type
GitOrigin-RevId: 57ba0633c7
|
4 years ago |