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
|
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
|
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
|
6e9a104d7b
|
fix(mge/cmake): fix cmake for opencl and add some minor fixes
GitOrigin-RevId: 9ea1c81407
|
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 |
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
|
32cd9ca0bd
|
fix(mgb): remove unnecessary null pointer checks
GitOrigin-RevId: 85aa0a9bdb
|
5 years ago |
Megvii Engine Team
|
4abe026b68
|
Merge branch 'master' into dev
GitOrigin-RevId: 4692108758
|
5 years ago |
Megvii Engine Team
|
044695aa2a
|
feat(mgb/plugin): add profiler for dynamic graph
GitOrigin-RevId: 10a6cccff9
|
5 years ago |
Megvii Engine Team
|
8b0315b3b1
|
fix(mgb): fix nhwcd4 optpass
GitOrigin-RevId: 9295abec77
|
5 years ago |
Megvii Engine Team
|
765c9d354c
|
fix(mgb/trt): change log level of trt log
GitOrigin-RevId: 5f5495e303
|
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
|
b8d27934cd
|
fix(mgb/core): fix opr shallow copy in sublinear
retrive comp_node from output var instead of opr config in sublinear if all outputs were placed on same comp_node
GitOrigin-RevId: a1ed027dac
|
5 years ago |
Megvii Engine Team
|
075c4562bc
|
fix(mgb/core): fix stream propagation in seq_comp_node_opt
for ForwardInToOutput op, StreamPropType of output(0) should be forwarded as input(0)
GitOrigin-RevId: 13e9587961
|
5 years ago |
Megvii Engine Team
|
fc6aa12e2f
|
feat(mge):add qint4/quint4 to python
GitOrigin-RevId: f94609db00
|
5 years ago |
Megvii Engine Team
|
a94fb7b126
|
fix(mgb/gopt): fix convert format opt pass by ensuring the replacement won't change output var's channel
GitOrigin-RevId: 170994935a
|
5 years ago |
Megvii Engine Team
|
20450817da
|
fix(mge/sdk): fix README.md and add missed example
GitOrigin-RevId: 0836029312
|
5 years ago |
Megvii Engine Team
|
5fe7cd1457
|
fix(mge/jit): fix typo InternalGraphGenerator
GitOrigin-RevId: 1ab14acbd2
|
5 years ago |
Megvii Engine Team
|
862de28a64
|
build(mgb): fix linkage of libatomic
GitOrigin-RevId: c164ed28e1
|
5 years ago |
Megvii Engine Team
|
a7e28712b5
|
fix(mgb): print verbose logs as debug log in TensorRT logger
GitOrigin-RevId: e197d565cb
|
5 years ago |
Megvii Engine Team
|
f91881ffdc
|
MegEngine: Initial commit of MegEngine.
GitOrigin-RevId: f0c8338beb
|
5 years ago |