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
Megvii Engine Team
5121626e98
refactor(dnn/parampack): remove parampacksplit
GitOrigin-RevId: ed00341d58
5 years ago
Megvii Engine Team
37b67c9b2b
refactor(dnn/parampack): reduce param pack memory use
GitOrigin-RevId: a802a14e8d
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
99fae95e02
feat(mge/parampack): add user-defined key to pack params
GitOrigin-RevId: 7d51dcae23
5 years ago
Megvii Engine Team
0668b343f0
feat(dnn/arm_common): add arm_common int8 channel wise stride 1
GitOrigin-RevId: 679916a986
5 years ago
Megvii Engine Team
6e9a104d7b
fix(mge/cmake): fix cmake for opencl and add some minor fixes
GitOrigin-RevId: 9ea1c81407
5 years ago
Megvii Engine Team
9a42c63641
fix(mge/module): add the more warnings in load_state_dict
In the before, the information of miss matching and unused operators won't be printed in the non-trict mode, this commit add the information.
GitOrigin-RevId: b2543eb832
5 years ago
Megvii Engine Team
9b74e6fc88
fix(mge/random): add one assert in distribution.py
low must be less than high in uniform sampling
GitOrigin-RevId: 34c33a1ac9
5 years ago
Megvii Engine Team
fc4dedc0bc
fix(mge/random): add lower bound and higher bound for uniform sampling
GitOrigin-RevId: 2a2c56fd17
5 years ago
Megvii Engine Team
712b87c8c1
feat(mgb/core): add comp node for cambricon
add testcase for cambricon comp node
GitOrigin-RevId: 7794faa47f
5 years ago
Megvii Engine Team
856ef62721
feat(mgb/core): support copy DeviceTensorND from cpu to cuda
GitOrigin-RevId: d56f4ebf1f
5 years ago
Megvii Engine Team
ca811c2c91
fix(mgb/opr/reduce): fix grad function of reduce mean and add testcase
GitOrigin-RevId: 390854bb2f
5 years ago