Megvii Engine Team
786afef461
feat(build): install CMake config module and pkg-config descriptor
Also, upgrade to CMake 3.13.
The commit also contains significant refactors, as otherwise it is not
possible to properly export target `megengine` to
MegEngine-targets.cmake:
1. Optionally use system provided Flatbuffers.
2. Optionally use system provided MKL-DNN (Tested with Debian).
3. Refactor megbrain and megdnn targets into object libraries.
4. Set different path in BUILD_INTERFACE and INSTALL_INTERFACE of
various target_include_directories.
5. Specify PUBLIC/PRIVATE on various target_link_libraries.
GitOrigin-RevId: df118a879e
5 years ago
Megvii Engine Team
cc4e1dfd7c
feat(mgb/compnode): expose mem_status and try_coalesce_all_free_memory to python
GitOrigin-RevId: 0ec2d556b1
5 years ago
Megvii Engine Team
01092feb9b
feat(mgb): add PackAllReducePass
GitOrigin-RevId: 59c1b45393
5 years ago
Megvii Engine Team
c7e6c658fd
refactor(mge/distribute): use is_root (and rank) in stead of rank and root at collective comm
GitOrigin-RevId: dccdb71553
5 years ago
Megvii Engine Team
d7bb62cfa1
refactor(mgb): move mm_handler from python module into opr-mm
GitOrigin-RevId: f401ce8603
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
dbf7764012
fix(mge/mm_handler): fix unused var ins
GitOrigin-RevId: 199d084e8f
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
396653df5f
feat(mge/api): expose sublinear related parameters at mge api level
GitOrigin-RevId: 7a47f5d0d5
5 years ago
Megvii Engine Team
b8cbd4510b
fix(mgb/mm_server): return -1 when create_mm_server failed
GitOrigin-RevId: 117bb80c06
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
cec7fd3c5f
fix(mgb/build): fix multi-machine macro and add test_distributed
GitOrigin-RevId: cb1bfe8742
5 years ago
Megvii Engine Team
3fce8544ec
refactor(mgb/parampacksplit): remove input offsets on device
GitOrigin-RevId: b2bf3bf155
5 years ago
Megvii Engine Team
5121626e98
refactor(dnn/parampack): remove parampacksplit
GitOrigin-RevId: ed00341d58
5 years ago
Megvii Engine Team
b708f15d39
refactor(mgb/param_pack): use shared mem for param pack
GitOrigin-RevId: bc56f09037
5 years ago
Megvii Engine Team
f18259d73a
refactor(mge/zmqrpc): zmqrpc return real port or -1
GitOrigin-RevId: 878b63c601
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
dbe2b89331
fix(midout): fix brain opr midout 2/2 (also see a6aa1574)
fix extern_c_opr midout
GitOrigin-RevId: 7de4f650d1
5 years ago
Megvii Engine Team
fc6aa12e2f
feat(mge):add qint4/quint4 to python
GitOrigin-RevId: f94609db00
5 years ago
Megvii Engine Team
646829b802
ci(pytest): use common method install pip package
GitOrigin-RevId: 87162b8b56
5 years ago
Megvii Engine Team
9edce7e294
ci(python_test): run python test with cuda
GitOrigin-RevId: bc51c2e4d5
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