Megvii Engine Team
|
714cb232bb
|
feat(dnn): add gemv supports in conv1x1 for NCHW44 and NCHW44_DOT(aarch64 binary size grows 2KB)
GitOrigin-RevId: f8b6d7a1b7
|
5 years ago |
Megvii Engine Team
|
9cc669637e
|
refactor(sdk/extern-opr): add cpu threads config for MACE
GitOrigin-RevId: 350ca7d15a
|
5 years ago |
Megvii Engine Team
|
f4694669b1
|
fix(dnn/arm): add target os and arch for cross compile
GitOrigin-RevId: a711aceb35
|
5 years ago |
Megvii Engine Team
|
7c2782366b
|
feat(core): redact exception messages if logging is disabled
GitOrigin-RevId: 1d6bda2380
|
5 years ago |
Megvii Engine Team
|
6137adc05a
|
feat(core): move more exceptions to base exception
mgb::cg::GraphError
mgb::NumRangeChecker::Error
mgb::VarSanityCheck::Error
GitOrigin-RevId: 4d79e3629c
|
5 years ago |
Megvii Engine Team
|
49f02e71fc
|
chore(format): format megbrain/exception.h
GitOrigin-RevId: 744ed23b2d
|
5 years ago |
Megvii Engine Team
|
9667d93cfe
|
feat(core): internal changes to exception types
See comments.
GitOrigin-RevId: ecd3c22faa
|
5 years ago |
Megvii Engine Team
|
c8f3742f0e
|
fix(mge/doc): fix wrong doc for megengine doc
GitOrigin-RevId: a38dc8e102
|
5 years ago |
Megvii Engine Team
|
b8b000db3b
|
feat(dnn/fallback): fix fallback interface of weight preprocess
GitOrigin-RevId: ca860f487e
|
5 years ago |
Megvii Engine Team
|
cf3a55ce17
|
fix(mgb/opr-mm): remove PeerDesc from RemoteSend and RemoteRecv
GitOrigin-RevId: b7a7bbd0da
|
5 years ago |
Megvii Engine Team
|
d53dab2fb2
|
fix(build): export target megengine
GitOrigin-RevId: 6daf6b2648
|
5 years ago |
Megvii Engine Team
|
845f42a38b
|
fix(midout/naive/warp_perspective): fix Template Functions instantiation
GitOrigin-RevId: c487f3663d
|
5 years ago |
Megvii Engine Team
|
5fb07c9964
|
fix(dnn/x86): fix cmake error for build x86 gtest
GitOrigin-RevId: 7613190733
|
5 years ago |
Megvii Engine Team
|
763b57add7
|
fix(dnn/cuda): fix INTMAX overflow in warp_perspective_cuda
GitOrigin-RevId: d7354e74e2
|
5 years ago |
Megvii Engine Team
|
2e6e570dfe
|
feat(dnn/fallback): add armv7 im2col mk4-dot int8 and
nchw44 float 3x3 s2 fuse packb speed up about 10%
GitOrigin-RevId: 3f864cef1d
|
5 years ago |
Megvii Engine Team
|
457a1e010c
|
refactor(imperative): initial merge of xxx and megengine
GitOrigin-RevId: 48a48f5c7f
|
5 years ago |
Megvii Engine Team
|
a808f807c6
|
feat(externcopr/nvidaopticalflow): fix output and fix cu context issue on
diff thread, nvof loader outptr share with mgb ptr, upgrade lastest nvof sdk
GitOrigin-RevId: 6e003c6428
|
5 years ago |
Megvii Engine Team
|
1adb262ad4
|
fix(dnn/naive): fix midout for pooling
GitOrigin-RevId: 4edd99f3ec
|
5 years ago |
Megvii Engine Team
|
32d7f25b19
|
fix(dnn/naive): fix midout for relayout_format
GitOrigin-RevId: 6ff9e2280e
|
5 years ago |
Megvii Engine Team
|
f856b17085
|
fix(mge/functional): reshape bias to (1, out_features) in linear
GitOrigin-RevId: a15880b7fc
|
5 years ago |
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 |