Megvii Engine Team
|
a5fad7d07c
|
feat(dnn): add compile for riscv64
GitOrigin-RevId: fa0c163527
|
4 years ago |
Megvii Engine Team
|
478f2c51dc
|
feat(mge/jit): add trace/dump options
GitOrigin-RevId: 0f43c14599
|
4 years ago |
Megvii Engine Team
|
44d0b5daf5
|
feat(imperative): enable to() to copy to device
GitOrigin-RevId: f9caf17d24
|
4 years ago |
Megvii Engine Team
|
3e11d89415
|
fix(dnn/dump): add more info for dump CD4
GitOrigin-RevId: 5840afaacd
|
4 years ago |
Megvii Engine Team
|
d49a5c1be2
|
fix(mge/imperative): remove codes newly introduced in numpy 1.19
GitOrigin-RevId: 874350cbc7
|
4 years ago |
Megvii Engine Team
|
69a7c55f55
|
test(mge/function): fix test for new optimizer api
GitOrigin-RevId: 8ae7720fe6
|
4 years ago |
Megvii Engine Team
|
a66d4b8bb8
|
fix(mge/parampacksplit): fix param pack split mem forward
GitOrigin-RevId: 8c001b73ff
|
4 years ago |
Megvii Engine Team
|
bd3b9cb6d4
|
fix(mge/oprmm): fix grad for collective comm
GitOrigin-RevId: 8e28f46c90
|
4 years ago |
Megvii Engine Team
|
8d02d10483
|
refactor(mge/distributed): change bcast_params_ to bcast_list_
GitOrigin-RevId: 26b452a6b7
|
4 years ago |
Megvii Engine Team
|
c7acba41fc
|
refactor(mge/optimizer): refine gradmanager api, record = __enter__
GitOrigin-RevId: 5376177237
|
4 years ago |
Megvii Engine Team
|
8c482b6709
|
fix(mge/grad): make register_after_backward_callback private
GitOrigin-RevId: 8eb6c0e628
|
4 years ago |
Megvii Engine Team
|
66b6daf777
|
test(mge/optimizer): fix test for new optimizer api
GitOrigin-RevId: 482ee62652
|
4 years ago |
Megvii Engine Team
|
e9104ef157
|
fix(mge/parampack): fix copy stream, import cycle
GitOrigin-RevId: 673e11c5b6
|
4 years ago |
Megvii Engine Team
|
e283663a02
|
fix(mge/imperative): update tests to new optimizer api
GitOrigin-RevId: 3d06e3db3c
|
4 years ago |
Megvii Engine Team
|
b5016b9d29
|
feat(mge/parampack): add parampack in allreduce callback
GitOrigin-RevId: 73d53eeba1
|
4 years ago |
Megvii Engine Team
|
5ae89c799b
|
refactor(mgb/grad): place grad at param.grad
GitOrigin-RevId: fddeace402
|
4 years ago |
Megvii Engine Team
|
9faa32fcb7
|
fix(mge/imperative): fix grad callback
GitOrigin-RevId: 6f843b0106
|
4 years ago |
Megvii Engine Team
|
6d4fd9384d
|
fix(mge/optimizer): remove distributed optimizer
GitOrigin-RevId: 3e5d0612f0
|
4 years ago |
Megvii Engine Team
|
e50fa074cb
|
fix(mge/imperative): remove backward from optimizer
GitOrigin-RevId: ad6ad444fa
|
4 years ago |
Megvii Engine Team
|
6070266766
|
refactor(mge/grad_manager): refactor gradmanager, add allreduce callback
GitOrigin-RevId: 086e2871e8
|
4 years ago |
Megvii Engine Team
|
3f2eac2fe1
|
fix(mge/imperative): move functional/distributed.py to distributed/functional.py
GitOrigin-RevId: 30cf2f514b
|
4 years ago |
Megvii Engine Team
|
b3889938dc
|
feat(mge/examples): add trace & dump example of cifar10 quantization
GitOrigin-RevId: cfc5e3483a
|
4 years ago |
Megvii Engine Team
|
67859f04e1
|
fix(imperative): add __array__ and __array_wrap__ for tensorwrapper
GitOrigin-RevId: 87d4ab6c8e
|
4 years ago |
Megvii Engine Team
|
b1ab3646f5
|
feat(imperative): add tensor sanity check
GitOrigin-RevId: 27243978e3
|
4 years ago |
Megvii Engine Team
|
5aa19f3d4a
|
test(mge/imperative): add more testcases for function
GitOrigin-RevId: e0675994f1
|
4 years ago |
Megvii Engine Team
|
2a300415bc
|
fix(mge): remove a duplicated __init__ call in jit.trace
GitOrigin-RevId: d898f42fc4
|
4 years ago |
Megvii Engine Team
|
517c1ec007
|
fix(mge): disable dedup for io callback oprs
GitOrigin-RevId: 424d347d83
|
4 years ago |
Megvii Engine Team
|
939bf3b28d
|
fix(mge/imperative): impl hashable for nms and cond_take
GitOrigin-RevId: 56918db014
|
4 years ago |
Megvii Engine Team
|
8dc23e0fdf
|
fix(mge/functional): fix indexing_one_hot and remote_recv
GitOrigin-RevId: 00bdfb502b
|
4 years ago |
Megvii Engine Team
|
3bbfef3009
|
fix(mge/imperative): impl hashable for coll-comm
GitOrigin-RevId: 76ab16a89b
|
4 years ago |
Megvii Engine Team
|
76fa71573b
|
feat(dnn/cuda): add cutlass nchw4 convolution
GitOrigin-RevId: 93c9b212f4
|
4 years ago |
Megvii Engine Team
|
1f3f4abc38
|
fix(dnn): fix compile warnings
GitOrigin-RevId: 519a6c0c34
|
4 years ago |
Megvii Engine Team
|
a558d4a253
|
fix(mgb/atlas): remove unnessary setdevice
GitOrigin-RevId: 7492de1f01
|
4 years ago |
Megvii Engine Team
|
14ceda0dd3
|
ci(mge): fix Windows CI
symlink does not work on Windows
GitOrigin-RevId: 45505b8925
|
4 years ago |
Megvii Engine Team
|
1fe8a21299
|
fix(mge): fix sublinear memory in jit.trace
GitOrigin-RevId: 190a330a8c
|
4 years ago |
Megvii Engine Team
|
2df1ab9633
|
refactor(mge/jit): skip seed when checking equal rng op
GitOrigin-RevId: dae2086b36
|
4 years ago |
Megvii Engine Team
|
5b6ebeb563
|
fix(mgb): append json file for dump and ready for midout open source
GitOrigin-RevId: 71ae7f1f4a
|
4 years ago |
Megvii Engine Team
|
a81abc1d28
|
fix(mgb): append json file for dump and ready for midout open source
GitOrigin-RevId: 71ae7f1f4a
|
4 years ago |
Megvii Engine Team
|
2beb65b19d
|
fix(mge/module): tensor shape will not work when constructing numpy array
GitOrigin-RevId: 5a0d705970
|
4 years ago |
Megvii Engine Team
|
651920c7ad
|
fix(dnn): fix nchw88 winograd weight preprocess
GitOrigin-RevId: 03c0c52aba
|
4 years ago |
Megvii Engine Team
|
adf5620e68
|
fix(dnn/x86): uniformity commit for dnnl
GitOrigin-RevId: 22362186dc
|
4 years ago |
Megvii Engine Team
|
5a378324b5
|
fix(whl/misc): easy whl local test, and fix export ALL_PYTHON at docker
env when value include more than two python version
GitOrigin-RevId: 8f1c0eb419
|
4 years ago |
Megvii Engine Team
|
fc3eca845b
|
feat(mge/imperative/jit): add dump input shape and xornet example
GitOrigin-RevId: 5e2acd4052
|
4 years ago |
Megvii Engine Team
|
1569cab0bc
|
feat(mge/imperative): add shufflenet example
GitOrigin-RevId: d29d8c2bd1
|
4 years ago |
Megvii Engine Team
|
087ceb52ef
|
feat(mge/imperative): add more optimizer trace tests
GitOrigin-RevId: 4127de1d22
|
4 years ago |
Megvii Engine Team
|
38a5c1c970
|
fix(mge): fix refcnt in TensorShape caster
GitOrigin-RevId: 24f6c7a6d0
|
4 years ago |
Megvii Engine Team
|
2cc85487d1
|
fix(imperative): fix hardcode of default device
GitOrigin-RevId: 722c4debfa
|
4 years ago |
Megvii Engine Team
|
403a1e7b24
|
feat(mge/imperative): add graph profiler
GitOrigin-RevId: c73563f337
|
4 years ago |
Megvii Engine Team
|
d06f248d90
|
fix(whl/mgb/imperative): fix symbols conflict runtime crash
GitOrigin-RevId: d60c4052c1
|
4 years ago |
Megvii Engine Team
|
1b2194cdce
|
feat(whl/internal): put imperative rt to megbrain
GitOrigin-RevId: f3b6e492d7
|
4 years ago |