Megvii Engine Team
|
47138c06cf
|
perf(dist): add fastpath for bcast params
GitOrigin-RevId: aa40b3cd72
|
4 years ago |
Megvii Engine Team
|
1a21dfdf5c
|
refactor(dtr): import dtr as submodule
GitOrigin-RevId: abecd0f176
|
4 years ago |
Megvii Engine Team
|
6964576e89
|
perf(dist): speed up bcast_val
GitOrigin-RevId: 21c4123b09
|
4 years ago |
Megvii Engine Team
|
052a600f03
|
feat(mge/module): add python wrapper for unfold
GitOrigin-RevId: 562103186f
|
4 years ago |
Megvii Engine Team
|
77a4bebff8
|
feat(mge/tools): add bash_completion for user to hint script
BREAKING CHANGE:
fix(mge/tools): change help script for other shell
BREAKING CHANGE:
fix(mge/tools): fix double -h error
BREAKING CHANGE:
fix(mge/tools): add support for _ parameter
BREAKING CHANGE:
GitOrigin-RevId: 537c3a2aae
|
4 years ago |
Megvii Engine Team
|
bff7d163a5
|
refactor(mge/jit): using static global_enable for apply ctx insted of global variable
GitOrigin-RevId: dd82b53faf
|
4 years ago |
Megvii Engine Team
|
2b7da42287
|
refactor(mge/jit): remove is_compiled flag in cpp tensor
GitOrigin-RevId: 15f90af735
|
4 years ago |
Megvii Engine Team
|
2762a55248
|
refactor(mge/jit): make trace return any kind of output
GitOrigin-RevId: fd1265c661
|
4 years ago |
Megvii Engine Team
|
a29bf679a8
|
refactor(mge/functional): remove dependence to trace in functional implementations
GitOrigin-RevId: 0b18479fcc
|
4 years ago |
Megvii Engine Team
|
2621c37801
|
feat(mge/tools): optimize statistical tools
BREAKING CHANGE:
GitOrigin-RevId: cd2a1acd11
|
4 years ago |
Megvii Engine Team
|
fb69697f6e
|
docs(mge/module): add note about weight shape in conv
GitOrigin-RevId: 43e1f15968
|
4 years ago |
Megvii Engine Team
|
186768e341
|
docs(misc): correct docstring format broadly
GitOrigin-RevId: 45234ca07e
|
4 years ago |
Megvii Engine Team
|
aa0e11febe
|
fix(imperative/dataset): fix VOC dataset
GitOrigin-RevId: a46a4c1173
|
4 years ago |
Megvii Engine Team
|
2c2beac915
|
fix(mge/utils): fix bug of VarNode inplace operations
GitOrigin-RevId: fa9eec7079
|
4 years ago |
Megvii Engine Team
|
4fa9096d20
|
feat(megbrain): add correlation opr
GitOrigin-RevId: 6d44598891
|
4 years ago |
Megvii Engine Team
|
866ab6e62e
|
fix(imperative/dataloader): collect garbage explicitly before starting a process in dataloader
GitOrigin-RevId: 3c3b51ad9c
|
4 years ago |
Megvii Engine Team
|
67167cb39d
|
feat(imperative/autodiff): add grad clip
GitOrigin-RevId: f344f4a233
|
4 years ago |
Megvii Engine Team
|
d4dd134d1a
|
refactor(mge/dtr): update dtr api
GitOrigin-RevId: dc366c65be
|
4 years ago |
Megvii Engine Team
|
7d18102795
|
fix(imperative/tensor): init m_offset when constructing a Tensor with DeviceTensorND
GitOrigin-RevId: b340e27c47
|
4 years ago |
Megvii Engine Team
|
45b9b5ed14
|
test(mge/distributed): add gather scatter reduce broadcast grad test
GitOrigin-RevId: 8245e11f1d
|
4 years ago |
Megvii Engine Team
|
78d1e2b2ca
|
fix(mge/distributed): fix gather scatter reduce broadcast autodiff
GitOrigin-RevId: 1c2250a079
|
4 years ago |
Megvii Engine Team
|
27ff38dd21
|
feat(functional/loss): add reduction choices to loss functions
GitOrigin-RevId: a29e6bb4cf
|
4 years ago |
Megvii Engine Team
|
0a1bd90f6b
|
test(param_pack): more cases for param pack concat
GitOrigin-RevId: 0700b548ab
|
4 years ago |
Megvii Engine Team
|
acb239927b
|
fix(mge/jit): error out if dump bn in training mode
GitOrigin-RevId: edc7ea2962
|
4 years ago |
Megvii Engine Team
|
dc12d319f6
|
fix(mge/network_visualize): fix warning 'span dist too large'
GitOrigin-RevId: 8db691b8ac
|
4 years ago |
Megvii Engine Team
|
e5d8e4279b
|
feat(mge): add a tool which can analyze the file generated by compare_binary_iodump.py
GitOrigin-RevId: 9acab0a49f
|
4 years ago |
Megvii Engine Team
|
a405461c01
|
fix(mge/elemwise): fix problem that elemwise.mode is not comparable with string mode
GitOrigin-RevId: 82e39be0a9
|
4 years ago |
Megvii Engine Team
|
53539fed01
|
fix(mge/quantization): fix a compute problem
GitOrigin-RevId: 8aaedea959
|
4 years ago |
Megvii Engine Team
|
94b5e1b046
|
fix(mge/core): support non-Tensor args for `core.tensor.utils.get_device`
GitOrigin-RevId: 51dc03db9e
|
4 years ago |
Megvii Engine Team
|
609b37b9ed
|
perf(fastrun): cache persistent cache
GitOrigin-RevId: 21f7d4c19d
|
4 years ago |
Megvii Engine Team
|
b6dc4c824d
|
fix(tensor): del valid tensors when compnode finalizing
GitOrigin-RevId: bace1f2b51
|
4 years ago |
Megvii Engine Team
|
1ecf2ab4ec
|
fix(mge/autodiff): check tensors to be attached
GitOrigin-RevId: c4f3f80876
|
4 years ago |
Megvii Engine Team
|
2c61b8bff4
|
feat(mge/functional): argmin and argmax support negtive axis
GitOrigin-RevId: a1bd1102a6
|
4 years ago |
Megvii Engine Team
|
cb84efa74f
|
fix(mge): fix dumping backward graph
GitOrigin-RevId: 430f110053
|
4 years ago |
Megvii Engine Team
|
bb94f2aa3d
|
feat(mge/module): add __repr__ method for qat and quantized module
GitOrigin-RevId: 0d78cbab9a
|
4 years ago |
Megvii Engine Team
|
5dbf3612b2
|
test(mge/optimizer): update optimizer test to make sure grad not change
GitOrigin-RevId: e207672116
|
4 years ago |
Megvii Engine Team
|
e9d6361e33
|
fix(mge/optimizer): fix optimizer update step inplace add grad
GitOrigin-RevId: d677d1ca6b
|
4 years ago |
Megvii Engine Team
|
a3e098c80c
|
fix(mge/utils): fix network multiple outputs issue
GitOrigin-RevId: d22e639cd3
|
4 years ago |
Megvii Engine Team
|
0d332cf005
|
feat(imperative/opr): rebase rng refactoring to dev & add python module
GitOrigin-RevId: ee5984c52d
|
4 years ago |
Megvii Engine Team
|
596dc982bb
|
feat(imperative): add rng opdef
GitOrigin-RevId: b62dcea7f5
|
4 years ago |
Megvii Engine Team
|
800f119c2e
|
feat(mge/tools): module_status-add-functions
BREAKING CHANGE:
GitOrigin-RevId: ced3da3a12
|
4 years ago |
Megvii Engine Team
|
3ef4fbdc36
|
fix(mge/module): fix a mistaken in BN docs
GitOrigin-RevId: 086a70649e
|
4 years ago |
Megvii Engine Team
|
6d08be373a
|
fix(data): fix the sampling scale in Lighting
GitOrigin-RevId: d0a1ca50e2
|
4 years ago |
Megvii Engine Team
|
11de49b943
|
feat(mge/module): python wrapper for conv_transpose3d
GitOrigin-RevId: 61097b8713
|
4 years ago |
Megvii Engine Team
|
331995e7f6
|
refactor(imperative): alloc enum type class on heap
GitOrigin-RevId: d2b2acea22
|
4 years ago |
Megvii Engine Team
|
cfbd4f9bf1
|
fix(device): fix get_device_count for xpu
GitOrigin-RevId: 6c77f507f3
|
4 years ago |
Megvii Engine Team
|
3bfe87ecb5
|
perf(tensor): always copy to pinned memory before h2d
GitOrigin-RevId: f945f2435f
|
4 years ago |
Megvii Engine Team
|
45a031fec3
|
test(dist): remove redundant assert for reduce_sum and gather
GitOrigin-RevId: bc7393bb63
|
4 years ago |
Megvii Engine Team
|
04b96602ae
|
feat(opr-mm): add backend argument for remote send/recv
GitOrigin-RevId: 841a0e45ab
|
4 years ago |
Megvii Engine Team
|
d4eb389042
|
feat(distributed): auto detect device and backend when init group
GitOrigin-RevId: 90be2d5b4d
|
4 years ago |