Megvii Engine Team
486cbdea8b
fix(mgb/opt): nchw to nchw4 pass suppport ic less than 4
GitOrigin-RevId: a3c205f38f
5 years ago
Megvii Engine Team
1c3d1f8602
fix(dnn): fix Image2DPack4TensorFormat check
GitOrigin-RevId: b9a8ae4e1a
5 years ago
Megvii Engine Team
7886ff9af0
feat(dnn): add relayout_format for nchw to nchw4 and ic <=4
GitOrigin-RevId: 07f2ee6c5b
5 years ago
Megvii Engine Team
1630a635d1
fix(dnn/native): also fix native logic
GitOrigin-RevId: a80f090271
5 years ago
Megvii Engine Team
6078187e32
fix(dnn/cuda): fix indexing logic in psroi_pooling
a variable relating to indexing was not computed correctly
GitOrigin-RevId: 548c8f3f14
5 years ago
Megvii Engine Team
32d91d5e6b
fix(dnn/arm_common): fix and optimize workspacebundle copy when algo compute
GitOrigin-RevId: 801aedbd72
5 years ago
Megvii Engine Team
caf77d000e
feat(mge/module): add quantize dtype load support for module load_state_dict
GitOrigin-RevId: 0a94cb6b17
5 years ago
Megvii Engine Team
dedb7a3f14
feat(dnn/cuda): add cuda remap
GitOrigin-RevId: 40a2a2ce24
5 years ago
Megvii Engine Team
cb5c5fd5dc
feat(mgb/opt): add nchw->nchw4 for tensorrt replace pass
GitOrigin-RevId: db114549be
5 years ago
Megvii Engine Team
2e70cf1dd0
feat(mgb/opt): add nchw->nchw4 in tensorcore pass
GitOrigin-RevId: 755f8dfefe
5 years ago
Megvii Engine Team
1e8337f14a
fix(mgb/gopt): remove redundant reshape in nchw->nchw4 pass
GitOrigin-RevId: 0f5c7c3e48
5 years ago
Megvii Engine Team
946a340c3d
feat(ci/midout): opt midout and add midout ci
GitOrigin-RevId: 1e5fe75255
5 years ago
Megvii Engine Team
ef437f69aa
fix(mge/data/dataloader): add refcount in _PlasmaStoreManager
GitOrigin-RevId: 9a95cb1a5d
5 years ago
Megvii Engine Team
90aa65d98d
fix(mgb/serialization): fix multiple graph load error
GitOrigin-RevId: 89414b014b
5 years ago
Megvii Engine Team
2b3a88d011
feat(mge/distributed): add parameter replica_mode
GitOrigin-RevId: 244e4ca437
5 years ago
Megvii Engine Team
44c381b6f4
Revert "feat(dnn/naive): workspacebundle support 2D"
This reverts commit 4408bb9e1d
.
GitOrigin-RevId: b5b23a8aae
5 years ago
Megvii Engine Team
cdbe44f8b4
feat(dnn): add gemv supports in conv1x1 with format NCHW
GitOrigin-RevId: 97679e8526
5 years ago
Megvii Engine Team
6972fc7d75
feat(opr/standalone): import NMSKeep from MegSkull contrib
1. clang-format.
2. Use name "NMSKeepMGB" during serialization to avoid conflict with
MegSkull contrib one.
GitOrigin-RevId: 627c60e482
5 years ago
Megvii Engine Team
5fa444c39a
refactor(mge/quantization): add `narrow_range` to control quant dtype's lower bound
GitOrigin-RevId: 92389341be
5 years ago
luzzyzhang
3b6285d7ba
ci(env): add nccl env
5 years ago
Megvii Engine Team
80ecabe8c6
feat(dnn/arm_common): add nchw_nchw44 aarch64 int8 3x3s2 7x7s2 asm
GitOrigin-RevId: 871465335d
5 years ago
Megvii Engine Team
6d367454cf
feat(mge/opr-mm): add param local_grad for collective_comm opr
GitOrigin-RevId: cc120cfb55
5 years ago
Megvii Engine Team
0ccb965c8e
fix(mgb/gopt): fix convert format nchw->nchw4 pass
GitOrigin-RevId: 1813753b14
5 years ago
Megvii Engine Team
1c2a525e2f
feat(imperative): set up imperative ci
GitOrigin-RevId: cc1b474abf
5 years ago
Megvii Engine Team
7ae05ac886
feat(imperative): merge common c++ code to megbrain
GitOrigin-RevId: d093778e10
5 years ago
Megvii Engine Team
9e904f683b
fix(dnn): fix can not inline small function with GCC compiler
GitOrigin-RevId: a23605c9e2
5 years ago
Megvii Engine Team
4ace67ff44
test(mge/distributed): check gpu num for multi gpu test
GitOrigin-RevId: 78f4718682
5 years ago
Megvii Engine Team
9d5c5c0788
feat(dnn/naive): workspacebundle support 2D
GitOrigin-RevId: 4408bb9e1d
5 years ago
Megvii Engine Team
f268e0f836
fix(dnn/x86): fix x86 compile in gcc 5.4
GitOrigin-RevId: b8683b1ad2
5 years ago
Megvii Engine Team
5d7fdd4762
fix(brain): fix winodws build
GitOrigin-RevId: 74eaea594e
5 years ago
megvii-mge
c36e5f8d4b
Update Help-wanted Issue.md
5 years ago
flashrunrun
8fdb3ea484
Create Help-wanted Issue.md
5 years ago
Xu Xinran
b532284cc2
chore(version): bump version.
5 years ago
Megvii Engine Team
8e9fa80cc2
feat(dnn/fallback): add matmul description for im2col
GitOrigin-RevId: 5bde0b60f0
5 years ago
Megvii Engine Team
af3de7e136
fix(jit): fix jit doc and add NCHW44_DOT
GitOrigin-RevId: 5f5feae8e7
5 years ago
Megvii Engine Team
7b8a99517c
feat(mge/utils): export _internal.plugin to mge.utils
GitOrigin-RevId: 795821fe7f
5 years ago
Megvii Engine Team
4e634d87e2
fix(load_and_run): fix dump_with_testcase_mge.py with import megbrain
GitOrigin-RevId: 5c1992fb37
5 years ago
Megvii Engine Team
318f6d75fa
fix(mge): fix some warnings
GitOrigin-RevId: 38b285f991
5 years ago
Megvii Engine Team
b9b531184f
fix(load_and_run): fix load_and_run with --input which ignore iters
GitOrigin-RevId: 870d90900d
5 years ago
Megvii Engine Team
69fe5ab3b3
feat(dnn/cuda): add conv2d-sass-kernel
GitOrigin-RevId: f284d5a4ce
5 years ago
Megvii Engine Team
1a70842eac
fix(serialization): fix model compatibility
GitOrigin-RevId: 1310cdc182
5 years ago
Megvii Engine Team
6ed791bc86
feat(mge/quant): add TQT quant method
GitOrigin-RevId: 00b1616e73
5 years ago
Megvii Engine Team
328fb36f17
feat(mgb/opr-mm): add Scatter, Gather, AllToAll oprs
GitOrigin-RevId: f75169ecd6
5 years ago
Megvii Engine Team
3f51a6a05b
build(third_party): Update megray version
5 years ago
Megvii Engine Team
285d70cb66
fix(build): find_package(Threads) if not already done
GitOrigin-RevId: 10ecb51947
5 years ago
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
4d35397bdf
fix(dnn/fallback): fix conv1x1/im2col usable and fuse-conv-bias get fp32xfp32-->qint8 error
GitOrigin-RevId: 5a3bfedd8a
5 years ago
Megvii Engine Team
12dc36a6ab
feat(mgb/gopt): add interface to reproducible
GitOrigin-RevId: f341bea40b
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
7bcead7561
ci(docker): add build-arg option to setup yum and pip mirrors
GitOrigin-RevId: bb7aa875da
5 years ago