Megvii Engine Team
|
01092feb9b
|
feat(mgb): add PackAllReducePass
GitOrigin-RevId: 59c1b45393
|
5 years ago |
Megvii Engine Team
|
30b3d3aa3e
|
fix(dnn/gopt): add convolution nchw44-dot format gopt
GitOrigin-RevId: e8e1e96379
|
5 years ago |
Megvii Engine Team
|
d19a637959
|
fix(python_module): hold GIL during _get_serialized_dtype
GitOrigin-RevId: 6b7288bd0c
|
5 years ago |
Megvii Engine Team
|
1d7fcecab2
|
feat(mge/serialization): add map location
GitOrigin-RevId: 4b6d83365b
|
5 years ago |
Megvii Engine Team
|
90107b6dda
|
feat(mge/cgtools): add network vistior interface with optional pruning
GitOrigin-RevId: cfa69e3e83
|
5 years ago |
Megvii Engine Team
|
45e2beead6
|
feat(mgb/gopt): add nchw4 optpass
GitOrigin-RevId: 551b6b828d
|
5 years ago |
Megvii Engine Team
|
e080dd3c8c
|
refactor(gopt): rename nchw2xxx to xxx
GitOrigin-RevId: fcb08c09e0
|
5 years ago |
Megvii Engine Team
|
a3560fa10e
|
feat(gopt): add tranform to chwn4 to optimize_for_inference
GitOrigin-RevId: 4d1a9c6c84
|
5 years ago |
Megvii Engine Team
|
1fb7d34f0b
|
refactor(gopt): refactor layout transform for optimize for inference
GitOrigin-RevId: df4e8866d3
|
5 years ago |
Megvii Engine Team
|
0293d58ade
|
feat(mge): add bfloat16 support
GitOrigin-RevId: a942ce6791
|
6 years ago |
Megvii Engine Team
|
c3d5b61f1c
|
feat(mgb/opr): add property params for python operator class
GitOrigin-RevId: af6da0e0ac
|
5 years ago |
Megvii Engine Team
|
554ce352c8
|
feat(mgb/gopt): add nchw44 optpass
GitOrigin-RevId: dc38724558
|
5 years ago |
Megvii Engine Team
|
9b41321926
|
fix(mge/functional): support negative axis in math.py
GitOrigin-RevId: 75143a7309
|
5 years ago |
Megvii Engine Team
|
86a3445e8e
|
build(python_module): use consistent flag declaration in SWIG and c++
Include cmake generated megbrain_build_config.h in SWIG in order to remove usage of MGB_DEF.
GitOrigin-RevId: d3e841eaf6
|
5 years ago |
Megvii Engine Team
|
d74a4ee931
|
fix(mge/build): use consistent genfiles/megbrain_build_config.h
Modify get_dynamic_info to work with cmake changes
Also rename get_dynamic_info as get_mem_allocation_info
GitOrigin-RevId: 383affe388
|
5 years ago |
Megvii Engine Team
|
044695aa2a
|
feat(mgb/plugin): add profiler for dynamic graph
GitOrigin-RevId: 10a6cccff9
|
5 years ago |
Megvii Engine Team
|
862de28a64
|
build(mgb): fix linkage of libatomic
GitOrigin-RevId: c164ed28e1
|
5 years ago |
Megvii Engine Team
|
f91881ffdc
|
MegEngine: Initial commit of MegEngine.
GitOrigin-RevId: f0c8338beb
|
6 years ago |