Megvii Engine Team
|
809d5056cd
|
feat(mge/distributed): enable pt shm allreduce
GitOrigin-RevId: 1dd5a02a51
|
4 years ago |
Megvii Engine Team
|
8480302da8
|
fix(autograd): make higher order grad experimental
GitOrigin-RevId: 81e1eb0ebf
|
4 years ago |
Megvii Engine Team
|
72531f2b4a
|
test(autograd): add more tests for higher order grad
GitOrigin-RevId: 5fc308f87a
|
4 years ago |
Megvii Engine Team
|
522e556b83
|
feat(autodiff): support higher order grad
GitOrigin-RevId: 86390d2179
|
4 years ago |
Megvii Engine Team
|
5198b7834c
|
fix(mge/functional): fix expand_dims for scalar
GitOrigin-RevId: 253ea608f7
|
4 years ago |
Megvii Engine Team
|
933dd9a497
|
feat(mge/distributed): add cuda env check before forked thread
style(core/comp_node): reformat code
GitOrigin-RevId: 372452a8eb
|
4 years ago |
Megvii Engine Team
|
a5060a2bfe
|
feat(mgb/opr): add check_has_inf kernel and opr
GitOrigin-RevId: 0d042dbfce
|
4 years ago |
Megvii Engine Team
|
c64b1c94cd
|
feat(imperative/functional): add roll in functional
GitOrigin-RevId: ff630f1fe7
|
4 years ago |
Megvii Engine Team
|
aa4e847692
|
fix(interpreter): release gil when interpreter blocking
GitOrigin-RevId: c48e9efa5b
|
4 years ago |
Megvii Engine Team
|
d6079db1f6
|
feat(profiler): add memory flow format for profiler
GitOrigin-RevId: f0eab26398
|
4 years ago |
Megvii Engine Team
|
4286bb9f51
|
feat(profiler): add launch script for profiler
GitOrigin-RevId: fdfb60b149
|
4 years ago |
Megvii Engine Team
|
9d47c3babd
|
feat(profiler): imperative profiler support tracing
GitOrigin-RevId: b247472feb
|
4 years ago |
Megvii Engine Team
|
1d64792b41
|
refactor(profiler): detach profiler from interpreter
GitOrigin-RevId: f3954728d1
|
4 years ago |
Megvii Engine Team
|
90d3905799
|
feat(mge): add mge._exit
GitOrigin-RevId: e7f348a677
|
4 years ago |
Megvii Engine Team
|
54a4d70eb5
|
feat(src/serialization): add support of serializing metadata
GitOrigin-RevId: b563c94451
|
4 years ago |
Megvii Engine Team
|
62bd6c823b
|
feat(cmake/debug): misc for build
* add asan build option
* fix cpuinfo build opt level
* fix host release build with out debug info
* opt "fix lite bazel/cmake symbols MR"
* other misc build opt
GitOrigin-RevId: 6ca286e195
|
4 years ago |
Megvii Engine Team
|
3e4e4c4604
|
feat(mgb/jit): add graph_opt_config and jit_config interfaces
GitOrigin-RevId: 170d9eeab2
|
4 years ago |
Megvii Engine Team
|
355153e158
|
feat(mge/dtr): add DTR in computing graph
GitOrigin-RevId: 8941810319
|
4 years ago |
Megvii Engine Team
|
29f7cdb84a
|
fix(mgb/opr): correct nvof out shape computation
GitOrigin-RevId: 16bf086e92
|
4 years ago |
Megvii Engine Team
|
d8ee0d7b5c
|
fix(mge/distributed): fix the mutli dataloader test error
GitOrigin-RevId: 86c8925916
|
4 years ago |
Megvii Engine Team
|
e275dfeca1
|
feat(imperative/python): support pooling mode "average" for avg pool2d module
GitOrigin-RevId: 9fe442129f
|
4 years ago |
Megvii Engine Team
|
01354337a9
|
fix(mge/autodiff): fix incorrect handling of tuple dy
GitOrigin-RevId: beca8e3711
|
4 years ago |
Megvii Engine Team
|
d2673c5abf
|
fix(ci/windows): add windows cuda test
GitOrigin-RevId: 706be83032
|
4 years ago |
Megvii Engine Team
|
241b35a697
|
refactor(ops): remove BackwardGraph op
GitOrigin-RevId: eda20e5760
|
4 years ago |
Megvii Engine Team
|
8a918717c0
|
feat(mgb): add megengine inference
GitOrigin-RevId: 6ffec6b418
|
4 years ago |
Megvii Engine Team
|
b06b589960
|
feat(mgb): get static graph memory info
GitOrigin-RevId: f31745f8df
|
4 years ago |
Megvii Engine Team
|
0cf4ff702e
|
perf(dist): add fastpath for bcast params
GitOrigin-RevId: aa40b3cd72
|
4 years ago |
Megvii Engine Team
|
5e54821845
|
docs(syncbn): complete syncbn document
GitOrigin-RevId: 9b8d546f0b
|
4 years ago |
Megvii Engine Team
|
e417798f9e
|
fix(mge): correct pytype when calling apply from python
GitOrigin-RevId: 6abfa06ada
|
4 years ago |
Megvii Engine Team
|
3352fe7003
|
refactor(dtr): import dtr as submodule
GitOrigin-RevId: abecd0f176
|
4 years ago |
Megvii Engine Team
|
a7581986bc
|
perf(dist): speed up bcast_val
GitOrigin-RevId: 21c4123b09
|
4 years ago |
Megvii Engine Team
|
1525a02530
|
feat(mge/module): add python wrapper for unfold
GitOrigin-RevId: 562103186f
|
4 years ago |
Megvii Engine Team
|
d07ef83aa8
|
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
|
efc7f04fd3
|
refactor(mge/jit): using static global_enable for apply ctx insted of global variable
GitOrigin-RevId: dd82b53faf
|
4 years ago |
Megvii Engine Team
|
750bde9513
|
refactor(mge/jit): remove is_compiled flag in cpp tensor
GitOrigin-RevId: 15f90af735
|
4 years ago |
Megvii Engine Team
|
4f240ec2d3
|
refactor(mge/jit): make trace return any kind of output
GitOrigin-RevId: fd1265c661
|
4 years ago |
Megvii Engine Team
|
c6b552cf84
|
refactor(mge/functional): remove dependence to trace in functional implementations
GitOrigin-RevId: 0b18479fcc
|
4 years ago |
Megvii Engine Team
|
46d96478a1
|
feat(mge/tools): optimize statistical tools
BREAKING CHANGE:
GitOrigin-RevId: cd2a1acd11
|
4 years ago |
Megvii Engine Team
|
7f9d441cbb
|
docs(mge/module): add note about weight shape in conv
GitOrigin-RevId: 43e1f15968
|
4 years ago |
Megvii Engine Team
|
8be70d170c
|
docs(misc): correct docstring format broadly
GitOrigin-RevId: 45234ca07e
|
4 years ago |
Megvii Engine Team
|
53446bcea5
|
fix(imperative/dataset): fix VOC dataset
GitOrigin-RevId: a46a4c1173
|
4 years ago |
Megvii Engine Team
|
8e5bf94857
|
fix(mge/utils): fix bug of VarNode inplace operations
GitOrigin-RevId: fa9eec7079
|
4 years ago |
Megvii Engine Team
|
13b15fb08c
|
feat(megbrain): add correlation opr
GitOrigin-RevId: 6d44598891
|
4 years ago |
Megvii Engine Team
|
6d376623de
|
fix(imperative/dataloader): collect garbage explicitly before starting a process in dataloader
GitOrigin-RevId: 3c3b51ad9c
|
4 years ago |
Megvii Engine Team
|
259894250b
|
feat(imperative/autodiff): add grad clip
GitOrigin-RevId: f344f4a233
|
4 years ago |
Megvii Engine Team
|
601a33a84d
|
refactor(mge/dtr): update dtr api
GitOrigin-RevId: dc366c65be
|
4 years ago |
Megvii Engine Team
|
aea5ac1363
|
fix(mge/distributed): fix gather scatter reduce broadcast autodiff
GitOrigin-RevId: 1c2250a079
|
4 years ago |
Megvii Engine Team
|
b1bf193e4d
|
feat(functional/loss): add reduction choices to loss functions
GitOrigin-RevId: a29e6bb4cf
|
4 years ago |
Megvii Engine Team
|
579a3d7786
|
fix(mge/jit): error out if dump bn in training mode
GitOrigin-RevId: edc7ea2962
|
4 years ago |
Megvii Engine Team
|
68da973eec
|
fix(mge/network_visualize): fix warning 'span dist too large'
GitOrigin-RevId: 8db691b8ac
|
4 years ago |