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
Megvii Engine Team
ec9de227b6
fix(mgb/dnn): fix opencl local opr
GitOrigin-RevId: 38e0a658a2
5 years ago
Megvii Engine Team
f5833a5294
fix(dnn/cuda): fix cublas matmul on sm60
GitOrigin-RevId: 3fc0c30a23
5 years ago
luzzyzhang
6bd09b3888
ci: refactor run cpp test script
5 years ago
luzzyzhang
96742c19ae
ci: add github action
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
720b6b49d6
build(mge): copy file required by setup.py to build directory
GitOrigin-RevId: 0b853e809f
5 years ago
Megvii Engine Team
e8494f729b
fix(unittest): remove the torch dependency in test_conv
GitOrigin-RevId: 36dcab2b95
5 years ago
Megvii Engine Team
779ed45d74
docs(mge/parampack): refine parampack doc
GitOrigin-RevId: 1cff83c308
5 years ago
Megvii Engine Team
8e8794040d
chore(release): bump version.
GitOrigin-RevId: 979d8c247b
5 years ago
wenjuan
0a38906ba3
doc(mge): add openi msg
5 years ago
wenjuan
7f60f27af8
doc(mge): modify readme
5 years ago
Megvii Engine Team
7ed98dd50a
feat(mge/functional): add conv_transpose2d support
GitOrigin-RevId: 9ad87a4ea9
5 years ago
Megvii Engine Team
20332705d3
docs(mge/functional): refine tensor transpose doc
GitOrigin-RevId: 763d74d61e
5 years ago
Megvii Engine Team
979122622a
feat(dnn): add nchw44 q8 direct conv stride 1
GitOrigin-RevId: 2ae87ff691
5 years ago
Megvii Engine Team
944a125c68
fix(mgb): improve exception safety with smart pointers
GitOrigin-RevId: e08421ac29
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
b588d93eb0
fix(mge): fix fast-run cache mismatch
GitOrigin-RevId: a41150f71a
5 years ago
Megvii Engine Team
90ca85541e
feat(dnn/x86): add avx2 int8 stride1 chanwise multithread conv
GitOrigin-RevId: 8f310c3d13
5 years ago
Megvii Engine Team
0bdb64c5f9
ci(test): dump test time result to json file
GitOrigin-RevId: 0ed092a0e8
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
888895e971
refactor(mge/test): make test_parampack more stable
GitOrigin-RevId: d82230ea07
5 years ago
Megvii Engine Team
da52256891
fix(mge/data): process mnist without generate new files
GitOrigin-RevId: 44a697c3fe
5 years ago
Megvii Engine Team
538d3de9d2
feat(dnn/aarch64): add arm64 nchw44 mk matmul
GitOrigin-RevId: 698a11c3fd
5 years ago
Megvii Engine Team
5178648422
fix(mge/doc): fix batch_norm2d doc
GitOrigin-RevId: 769ddd367b
5 years ago
Megvii Engine Team
5e8aa33328
refactor(dnn): refactor winograd output transpose
GitOrigin-RevId: 6d4b225ea5
5 years ago
Megvii Engine Team
c6eb2e8d71
refactor(dnn): optimize winograd input transpose
GitOrigin-RevId: a43077550c
5 years ago
Megvii Engine Team
f077a5292c
fix(mge/param_pack): release old parameters
GitOrigin-RevId: 40a1f044e9
5 years ago
Megvii Engine Team
48822557b8
docs(mge/bn): fix docs and tests of batchnorm
GitOrigin-RevId: 8a96aa5fc2
5 years ago
Megvii Engine Team
2c4d1afe14
test(mge): move `NUMBER` config to pytest.init
GitOrigin-RevId: 1d82209c40
5 years ago
Megvii Engine Team
f33a92d635
docs(mge): fix api docs of loss and tranform
GitOrigin-RevId: 4d4d32b3c4
5 years ago
Megvii Engine Team
fd83e0252d
fix(mge/test): fix reproducibility of test_correctness.py
GitOrigin-RevId: 70bd43bbab
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
dc6f89f2d3
refactor(dnn): refactor winograd strategy helper
GitOrigin-RevId: ecc2b15df9
5 years ago
Megvii Engine Team
e1b2d31d36
test(mge): skip some doctests having different results on gpu
GitOrigin-RevId: 66e5db5c89
5 years ago
Megvii Engine Team
20b67b294c
ci(python): add format and lint check for python
GitOrigin-RevId: d8d106b927
5 years ago
Megvii Engine Team
62b0c6cd27
style(mge): apply format.sh
GitOrigin-RevId: a900b1bb6e
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
2c2caf3331
refactor(mge/data/dataloader): refactor the implementation of parallel dataloader
GitOrigin-RevId: 0554ee8427
5 years ago