Megvii Engine Team
|
d3730036ea
|
test(mge/dist): add unit tests for megengine.distributed.utils
GitOrigin-RevId: 32336c66f1
|
5 years ago |
Megvii Engine Team
|
ffe0409398
|
docs(mge/optimizer): refine the docstring of several apis
GitOrigin-RevId: a97fe5b68a
|
5 years ago |
Megvii Engine Team
|
0c54f2dc13
|
docs(mge/core): refine the docstring of several apis
GitOrigin-RevId: c03fecfa6a
|
5 years ago |
Megvii Engine Team
|
80d9e56b38
|
fix(mge/tensor): where support x or y is empty
GitOrigin-RevId: 2b69ee2e79
|
5 years ago |
Megvii Engine Team
|
8cf7150df0
|
feat(dnn/mge): add compnode multithread in python
GitOrigin-RevId: 47373d291d
|
5 years ago |
Megvii Engine Team
|
4997270156
|
docs(mge/module): refine the docstring of several apis
GitOrigin-RevId: ea04e05be4
|
5 years ago |
Megvii Engine Team
|
4c5d01fa34
|
refactor(mge/examples): add example to use modified sublinear API
GitOrigin-RevId: c6f6c65d0b
|
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
|
c985204b31
|
feat(dnn): add conv1x1 algo and tests
GitOrigin-RevId: 374a62cf12
|
5 years ago |
Megvii Engine Team
|
d8d3f405bb
|
fix(mge/data): fix an nan bug in coco dataset
GitOrigin-RevId: 689fbaafa5
|
5 years ago |
Megvii Engine Team
|
41464e1818
|
feat(mge/data): voc dataset supports detection
GitOrigin-RevId: f78bef3cd2
|
5 years ago |
Megvii Engine Team
|
26738d99eb
|
feat(mge/data): add objects365 dataset
GitOrigin-RevId: 8eccebfc1d
|
5 years ago |
Megvii Engine Team
|
aa2bfd2d85
|
fix(mge/data): create dir automatically when download=True
GitOrigin-RevId: f3cb1b7d50
|
5 years ago |
Megvii Engine Team
|
3c6163868b
|
fix(dnn): add cmake support for midout
GitOrigin-RevId: 58fcbed88a
|
5 years ago |
Megvii Engine Team
|
d345c86277
|
feat(dnn/fallback): add im2col none dot int8 nchw44 support
GitOrigin-RevId: d326035202
|
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
|
b87a9a74b1
|
feat(mgb/opr-mm): add debug log for CollectiveComm opr
GitOrigin-RevId: 11bcf11043
|
5 years ago |
Megvii Engine Team
|
d6b098a052
|
refactor(dnn/cuda): remove function name weiming
GitOrigin-RevId: f36495a46a
|
5 years ago |
Megvii Engine Team
|
ca24c4cdd8
|
refactor(dnn/cuda): invoke local opr in group local
GitOrigin-RevId: aa636041bb
|
5 years ago |
Megvii Engine Team
|
91c3d5fe15
|
fix(dnn/cuda): add sharemem to prefer and refactor code
GitOrigin-RevId: 0dba39f56b
|
5 years ago |
Megvii Engine Team
|
abf82cfb50
|
feat(mge/dtype): export `_metadata_dict` for consistent dtype property
GitOrigin-RevId: 6840c0b6b4
|
5 years ago |
Megvii Engine Team
|
f582c1921e
|
fix(dnn/test): fix dnn assert and format code
GitOrigin-RevId: c684210c42
|
5 years ago |
Megvii Engine Team
|
9cc7b32e6d
|
refactor(dnn/arm): refactor arm_common pooling for add nchw44 pooling
GitOrigin-RevId: 3980ceddee
|
5 years ago |
Megvii Engine Team
|
dd962a4700
|
fix(mge/module): add negative slope attribute for LeakyReLU module
GitOrigin-RevId: 67e1967b56
|
5 years ago |
Megvii Engine Team
|
64401dfef0
|
fix(dnn/x86): fix gcc5 -O0 compiler error
GitOrigin-RevId: 237e2f27f2
|
5 years ago |
Megvii Engine Team
|
e3de2011c0
|
build(mge): clang compile and link error
GitOrigin-RevId: 7062b451aa
|
5 years ago |
Megvii Engine Team
|
554ce352c8
|
feat(mgb/gopt): add nchw44 optpass
GitOrigin-RevId: dc38724558
|
5 years ago |
Megvii Engine Team
|
7d1e1f9ae9
|
feat(mgb/gopt): split inference.cpp into two file
GitOrigin-RevId: d32f30d313
|
5 years ago |
Megvii Engine Team
|
5507a29bd3
|
feat(mge/functional): add normalize opr
GitOrigin-RevId: 572a32f563
|
5 years ago |
Megvii Engine Team
|
000663c30c
|
feat(mge/functional): add isnan and isinf oprs
GitOrigin-RevId: b4a347751c
|
5 years ago |
Megvii Engine Team
|
ccc95ad2a9
|
fix(mgb/gopt): fix remove redundant typecvt pass
GitOrigin-RevId: 6a7957e362
|
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
|
207527d108
|
fix(mge/clamp): fix `F.clamp`
GitOrigin-RevId: 1efac8add6
|
5 years ago |
Megvii Engine Team
|
8ac7333378
|
fix(dnn): fix data type check for quantized convolution
GitOrigin-RevId: e0f97052ff
|
5 years ago |
Megvii Engine Team
|
c59be192cd
|
feat(dnn/arm_common/elemwise): add arm_common support chw44 elemwise
GitOrigin-RevId: aba44e0123
|
5 years ago |
Megvii Engine Team
|
6ffcfb4ce2
|
refactor(mge/dataloader): make plasma_store starting more robust
GitOrigin-RevId: e1f78db99a
|
5 years ago |
Megvii Engine Team
|
920eeeb56f
|
fix(mge/build): remove MEGDNN flags from command line
GitOrigin-RevId: a9813a44e4
|
5 years ago |
Xinran Xu
|
a6253fb7c4
|
build: remove useless files
|
5 years ago |
luzzyzhang
|
8fcc072788
|
ci(workflow): add gpu ci workflow
|
5 years ago |
luzzyzhang
|
16f052e916
|
fix(megdnn): change ver 60 to use cuda capability 50
fix(megdnn): if compute capability less than 60 skip fp16 test
|
5 years ago |
Megvii Engine Team
|
c4dfdbd26d
|
refactor(dnn/fallback): refactor im2col
GitOrigin-RevId: b58770211e
|
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
|
e729006f8e
|
docs(mge/python): add doc for zero_grad
GitOrigin-RevId: bb5154c0e7
|
5 years ago |
Megvii Engine Team
|
bb92ee26e3
|
fix(mge/functional): refine doc and add test for linspace
GitOrigin-RevId: d6c7eea2c2
|
5 years ago |
Megvii Engine Team
|
ce30d045ee
|
feat(mge/functional): add arange in tensor.py
GitOrigin-RevId: ad88a4c18e
|
5 years ago |
Megvii Engine Team
|
23478a0d53
|
test(dnn/cuda): fix cuda int8 test on sm60
GitOrigin-RevId: 66bab333e1
|
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
|
11388e5888
|
refactor(mgb/parampacksplit): remove m_mem_plan_fwd
GitOrigin-RevId: 2e624b79d5
|
5 years ago |
Megvii Engine Team
|
d929dd0a8d
|
docs(mgb/parampack): add description of parampack opr param
GitOrigin-RevId: 485290aa09
|
5 years ago |