Megvii Engine Team
|
0bdd0b1467
|
refactor(dispatch): switch to new dispatch system
GitOrigin-RevId: 32dd49a23a
|
3 years ago |
Megvii Engine Team
|
42759dc7f5
|
test(dtr): make dtr_resnet1202 isolated
GitOrigin-RevId: 4d863cc32e
|
3 years ago |
Megvii Engine Team
|
89ed7ab2ff
|
test(imperative): speed up dtr test
GitOrigin-RevId: 57f092d729
|
3 years ago |
Megvii Engine Team
|
d909950f64
|
refactor(imperative): remove swap in dtr
GitOrigin-RevId: 5c9e42f74a
|
3 years ago |
Megvii Engine Team
|
2c5a007a92
|
fix(mge/optimizer): allow lr to be 0
GitOrigin-RevId: dabd1fcc33
|
3 years ago |
Megvii Engine Team
|
4e95c13617
|
feat(sgd): sgd supports nesterov momentum
GitOrigin-RevId: 13eda179da
|
3 years ago |
Megvii Engine Team
|
dc28a43409
|
fix(mgb/bn): fix empty tensor input problem and other minor problems
GitOrigin-RevId: 1708b76cb8
|
3 years ago |
Megvii Engine Team
|
27f2ecc436
|
feat(imperative/opr): add extern c opr and support dump
GitOrigin-RevId: b29e374c60
|
3 years ago |
Megvii Engine Team
|
e5217db9f9
|
feat(traced_module): move traced_module out of the experimental folder
GitOrigin-RevId: 36c76b5277
|
3 years ago |
Megvii Engine Team
|
f88bd3ae33
|
refactor(traced_module): let TracedModule own argdef_graph_map
GitOrigin-RevId: 80d685b9a3
|
3 years ago |
Megvii Engine Team
|
442b4f6c26
|
test(traced_module): add some testcases for traced module
GitOrigin-RevId: 0d6bb20b2b
|
4 years ago |
Megvii Engine Team
|
d86ed426ee
|
fix(dtr): simulate the system stack to avoid stack overflow during recomputing
GitOrigin-RevId: cb73e62b19
|
3 years ago |
Megvii Engine Team
|
207a346351
|
chore(mge): run get_device_count("gpu") in subprocess
GitOrigin-RevId: 0f0dc001cf
|
4 years ago |
Megvii Engine Team
|
e6caa9ff89
|
feat(opr): add bn backward for inference mode
GitOrigin-RevId: bb643cb62f
|
4 years ago |
Megvii Engine Team
|
070c811732
|
fix(imperative): remove convert_inputs
GitOrigin-RevId: a3c43db746
|
3 years ago |
Megvii Engine Team
|
d6db4fea93
|
fix(mge/module): set no_cache=true when loading state dict
GitOrigin-RevId: 83281a3d47
|
4 years ago |
Megvii Engine Team
|
c78a7848a1
|
test(profiler): simple tests for profiler
GitOrigin-RevId: 92073ce357
|
4 years ago |
Megvii Engine Team
|
1d64792b41
|
refactor(profiler): detach profiler from interpreter
GitOrigin-RevId: f3954728d1
|
4 years ago |
Megvii Engine Team
|
259894250b
|
feat(imperative/autodiff): add grad clip
GitOrigin-RevId: f344f4a233
|
4 years ago |
Megvii Engine Team
|
dadd508615
|
test(param_pack): more cases for param pack concat
GitOrigin-RevId: 0700b548ab
|
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
|
9451a96147
|
test(mge/optimizer): update optimizer test to make sure grad not change
GitOrigin-RevId: e207672116
|
4 years ago |
Megvii Engine Team
|
fe99cdc794
|
feat(interpreter): add dynamic sublinear
GitOrigin-RevId: 7de54fe743
|
4 years ago |
Megvii Engine Team
|
7e22e9f06e
|
feat(optimzer): add AdamW
GitOrigin-RevId: e608b5d5b9
|
4 years ago |
Megvii Engine Team
|
04b1a45af4
|
fix(dnn): fix cudnn crash when finalize called after cudnn dtor
GitOrigin-RevId: b0ad639921
|
4 years ago |
Megvii Engine Team
|
31e4bf2c34
|
fix(mge/optimizer): fix optimizer's state_dict bug
GitOrigin-RevId: 67fb112fb8
|
4 years ago |
Megvii Engine Team
|
a3ea1f153c
|
feat(mgb/opr): add fast profile and combined Execution strategy
GitOrigin-RevId: 843dc3a790
|
4 years ago |
Megvii Engine Team
|
bb8f292816
|
refactor(mge/func): change `conv_execution_strategy` to `execution_strategy`
GitOrigin-RevId: 7aef9935ee
|
4 years ago |
Megvii Engine Team
|
30d6b4f65d
|
fix(mge): fix scalar parameter change to 1-dim parameter after backward and optimize
GitOrigin-RevId: 1794369a71
|
4 years ago |
Megvii Engine Team
|
649e4dd750
|
test(cuda): fix test for cu111
GitOrigin-RevId: 04fe5eb23f
|
4 years ago |
Megvii Engine Team
|
e82fa4ec23
|
fix(gopt): using new_inp for build_chain in DelayBroadcast pass
GitOrigin-RevId: efc6377197
|
4 years ago |
Megvii Engine Team
|
43b4d4a430
|
feat(imperative/test): add more correctness tests
GitOrigin-RevId: 2ef6027bb1
|
4 years ago |
Megvii Engine Team
|
dbb3dd681f
|
refactor(profiler): integrate profiler into interpreter
GitOrigin-RevId: ccc984acbd
|
4 years ago |
Megvii Engine Team
|
ff05667b48
|
test(dist): refactor distributed test with fixtures
GitOrigin-RevId: e69acb72da
|
4 years ago |
Megvii Engine Team
|
27a9326c66
|
fix(mge/module): fix frozen batch norm
GitOrigin-RevId: 143d468a37
|
4 years ago |
Megvii Engine Team
|
364afec033
|
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
|
4 years ago |
Megvii Engine Team
|
b83c77e105
|
test(mge/test_save_load): remove test model file
GitOrigin-RevId: 893b82b1c9
|
4 years ago |
Megvii Engine Team
|
c49427d15a
|
feat(imperative): add inplace add_update option in optimizer
GitOrigin-RevId: b8feb49321
|
4 years ago |
Megvii Engine Team
|
b9762d714c
|
fix(mge): make parampack run with tensor symbolic shape
GitOrigin-RevId: 6fc313785d
|
4 years ago |
Megvii Engine Team
|
60c7d62a4c
|
refactor(imperative): remove multidispatch, raw_tensor, register
GitOrigin-RevId: ca5a6ed8eb
|
4 years ago |
Megvii Engine Team
|
c294b9d18b
|
refactor(mge/tensor): remove old implementation
remove core.tensor, raw_tensor,TensorWrapper
avoid create tensor with zero-stride numpy ndarray
GitOrigin-RevId: 4fe5c4c5ba
|
4 years ago |
Megvii Engine Team
|
15e8e7df76
|
refactor(mge/imperative): migrate to new core implementation
* swap/drop
* config/set_async_level
* _dev_tensor
* sync
GitOrigin-RevId: 850fb98852
|
4 years ago |
Megvii Engine Team
|
9d928e7f83
|
refactor(mge/distributed): sync interpreter for distribtued launcher
GitOrigin-RevId: 8a88c272a1
|
4 years ago |
Megvii Engine Team
|
ee4ea7fdc8
|
test(distributed/test): make distributed test more stronger
GitOrigin-RevId: 085fd1dcfd
|
4 years ago |
Megvii Engine Team
|
7aa54b0ec6
|
feat(mge): enable memory swap and drop/recomputation
GitOrigin-RevId: c56c87c88b
|
4 years ago |
Megvii Engine Team
|
638ab52fdc
|
feat(mge/imperative): simulates scalar
GitOrigin-RevId: e81630e256
|
4 years ago |
Megvii Engine Team
|
94dba16ff4
|
perf(mge/imperative): misc optimizations
GitOrigin-RevId: bbe7a10b00
|
4 years ago |
Megvii Engine Team
|
cc85047bf0
|
fix(mge/trace): fix sublinear in trace
GitOrigin-RevId: 356dcd9523
|
4 years ago |
Megvii Engine Team
|
cae8c8a4a4
|
test(mge/parampack): add test for parampack
GitOrigin-RevId: f3e423dffc
|
4 years ago |
Megvii Engine Team
|
7cd846a570
|
fix(mge/batch_norm): fix batch_norm check when trace(symbolic=True)
GitOrigin-RevId: 2032eb5f7d
|
4 years ago |