Megvii Engine Team
|
bd817f3af2
|
perf(syncbn): fallback to bn when sync is not required
GitOrigin-RevId: e89fdd87b0
|
3 years ago |
Megvii Engine Team
|
ffcb4dacd9
|
style(mge): change the pad opr doc and format
GitOrigin-RevId: 7b20e96519
|
3 years ago |
Megvii Engine Team
|
43cb305e53
|
feat(mge): add imperative pad
GitOrigin-RevId: de79de536f
|
3 years ago |
Megvii Engine Team
|
0a665ea452
|
feat(mge/device): enable to get cuda compute capability
GitOrigin-RevId: b5d3f2225c
|
3 years ago |
Megvii Engine Team
|
7a9f2ed95e
|
perf(functional/split): add python binding for Split opr
GitOrigin-RevId: 3359f5da26
|
3 years ago |
Megvii Engine Team
|
ab309eb5fc
|
feat(mgb/opr): let Split support empty IO
GitOrigin-RevId: aad6dc06bf
|
3 years ago |
Megvii Engine Team
|
8485eff12c
|
docs(mge/functional): add docs for megengine.functional.sub
GitOrigin-RevId: 40208739bd
|
3 years ago |
Megvii Engine Team
|
e6a8b0256e
|
fix(mge): ignore errors caused by earlier async errors
GitOrigin-RevId: ce2028d38a
|
3 years ago |
Megvii Engine Team
|
ee790cb62c
|
fix(imperative/python): fix warp_perspective doc and arange dtype
GitOrigin-RevId: dc4e021ec0
|
3 years ago |
Megvii Engine Team
|
c5f8b58385
|
docs(docstring): transfer to google style
GitOrigin-RevId: a71245c553
|
3 years ago |
Megvii Engine Team
|
76ce81e828
|
fix(mge): fix F.nn.dropout train and inference bugs
GitOrigin-RevId: 9d9f246d7b
|
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
|
03e80759b6
|
feat(mge/distributed): add hybird parallel Opr
GitOrigin-RevId: ff26671746
|
3 years ago |
Megvii Engine Team
|
e5217db9f9
|
feat(traced_module): move traced_module out of the experimental folder
GitOrigin-RevId: 36c76b5277
|
3 years ago |
Megvii Engine Team
|
6d1a4f20e7
|
feat(traced_module): support tracing submodules in list/dict
GitOrigin-RevId: 4076b47a89
|
3 years ago |
Megvii Engine Team
|
a3f9073c2c
|
feat(traced_module): update graph transform and add _module_name
GitOrigin-RevId: ef63ae0fd0
|
3 years ago |
Megvii Engine Team
|
b3d0affab1
|
feat(traced_module): support trace custom qat module
GitOrigin-RevId: 49f70a5f46
|
3 years ago |
Megvii Engine Team
|
15712807b9
|
feat(traced_module): add name to Node
GitOrigin-RevId: 39c2809067
|
3 years ago |
Megvii Engine Team
|
e918f0aa75
|
feat(traced_module): add treedef leaf node check and add some graph api
GitOrigin-RevId: 36c069bfee
|
3 years ago |
Megvii Engine Team
|
c7e730bc12
|
feat(traced_module): add some functions of graph modification
GitOrigin-RevId: 09691ebd33
|
3 years ago |
Megvii Engine Team
|
f88bd3ae33
|
refactor(traced_module): let TracedModule own argdef_graph_map
GitOrigin-RevId: 80d685b9a3
|
3 years ago |
Megvii Engine Team
|
b1c46ba46c
|
feat(traced_module): add some functions of graph modification
GitOrigin-RevId: ac0603057a
|
3 years ago |
Megvii Engine Team
|
4bb253695b
|
feat(traced_module): let CallFunction own graph
GitOrigin-RevId: 66cdbca7e5
|
3 years ago |
Megvii Engine Team
|
9a6a379346
|
feat(traced_module): add visit method
GitOrigin-RevId: 251ecebf87
|
4 years ago |
Megvii Engine Team
|
442b4f6c26
|
test(traced_module): add some testcases for traced module
GitOrigin-RevId: 0d6bb20b2b
|
4 years ago |
Megvii Engine Team
|
f2691566fd
|
feat(traced_module): add pytree
GitOrigin-RevId: 6c6e53521c
|
4 years ago |
Megvii Engine Team
|
bee305beb2
|
feat(traced_module): add functional trace and CallMethod/Function expr
GitOrigin-RevId: ad2cdc1b61
|
4 years ago |
Megvii Engine Team
|
763c56f3b9
|
feat(imperative): add traced module
GitOrigin-RevId: 28c3503f2e
|
4 years ago |
Megvii Engine Team
|
9279104b11
|
feat(mge): add opdef serialization and apply_module_trace
GitOrigin-RevId: 5b45bded1d
|
3 years ago |
Megvii Engine Team
|
29d24dbb80
|
fix(mge/function): fix interpolate unsupport fp16 error
GitOrigin-RevId: 7fc6271986
|
3 years ago |
Megvii Engine Team
|
e21967bb40
|
feat(mge): add env MGE_FASTRUN_CACHE_DIR
GitOrigin-RevId: 0351ff88d1
|
3 years ago |
Megvii Engine Team
|
6a1ec8a890
|
feat(mge): add git commit-id into fastrun cache key
GitOrigin-RevId: fb614b4d9c
|
3 years ago |
Megvii Engine Team
|
ae87876d34
|
feat(mge): refactor weightscaler
GitOrigin-RevId: 7f874388f7
|
3 years ago |
Megvii Engine Team
|
67575d582c
|
feat(mge/opr): add interpolate bilinear mode
GitOrigin-RevId: f7023a3fd3
|
3 years ago |
Megvii Engine Team
|
0558b2123d
|
feat(mge/opr): add interpolate nearest mode
GitOrigin-RevId: d384b87f50
|
3 years ago |
Megvii Engine Team
|
12cdbddd14
|
fix(ci): clean fastrun cache in windows and macos ci
GitOrigin-RevId: d1a010287f
|
3 years ago |
Megvii Engine Team
|
31705913c0
|
fix(ci): set MGE_FASTRUN_CACHE_TYPE=FILE in ci env
GitOrigin-RevId: c4a549480e
|
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
|
3206af9db2
|
perf(functional/matmul): reimplement matmul with subgraph
GitOrigin-RevId: 456b2a51d3
|
3 years ago |
Megvii Engine Team
|
8c47c1f149
|
perf(syncbn): reimplement with subgraph
GitOrigin-RevId: 13e7e3d3c0
|
3 years ago |
Megvii Engine Team
|
3103180456
|
fix(mge/optimizer): only disable convert inputs in build-in optimizers
GitOrigin-RevId: 1a48fe318d
|
3 years ago |
Megvii Engine Team
|
fd24dc8e36
|
feat(mge/hub/fetcher): add checking of mge version
GitOrigin-RevId: 808af7f82a
|
3 years ago |
Megvii Engine Team
|
77309609fa
|
perf(functional/dropout): add fastpath for dropout
GitOrigin-RevId: 3bf8546908
|
3 years ago |
Megvii Engine Team
|
92079e1061
|
docs(mge/distributed): update distributed collective communication Google style docs
GitOrigin-RevId: 871c4052b0
|
3 years ago |
Megvii Engine Team
|
5fe789abb9
|
fix(mge/distributed): skip check when machine_ranks is None
GitOrigin-RevId: a18ee052e9
|
3 years ago |
Megvii Engine Team
|
298c44433f
|
fix(mge/module): fix mistake warning message of module auto naming
GitOrigin-RevId: 4b85c13e90
|
3 years ago |
Megvii Engine Team
|
c54a731b9b
|
fix(utils/network): fix replace oprs
GitOrigin-RevId: eba27e3dfb
|
3 years ago |
Megvii Engine Team
|
18274e023c
|
fix(imperative): fix error message when applying custom function with non-tensor arguments
GitOrigin-RevId: 387d6fda4a
|
3 years ago |