Megvii Engine Team
|
d7c546c90a
|
fix(mge/interpreter): regenerates tensor when its dev value is needed
GitOrigin-RevId: ed26d52ee4
|
2 years ago |
Megvii Engine Team
|
2f829aaa7b
|
test(imperative): speed up integration test
GitOrigin-RevId: b03075fcfc
|
3 years ago |
Megvii Engine Team
|
c2e9860feb
|
chore(license): remove all license in file header
GitOrigin-RevId: a0e31247a6
|
3 years ago |
Megvii Engine Team
|
597a1e791b
|
refactor(imperative): add interface to clear algorithm cache
GitOrigin-RevId: 662618954b
|
3 years ago |
Megvii Engine Team
|
409c988163
|
fix(imperative): add matmul apply_on_varnode
GitOrigin-RevId: 2cf6bf237c
|
3 years ago |
Megvii Engine Team
|
4d5faa3f0a
|
fix(imperative): using DnnOprCaller to avoid early destruction of dnn_opr
GitOrigin-RevId: 4a14f53738
|
3 years ago |
Megvii Engine Team
|
a4327c4d25
|
perf(imperative): add dim_expansion transform for conv/bn1d
GitOrigin-RevId: d14a69424d
|
3 years ago |
Megvii Engine Team
|
0cb60d646d
|
feat(imperative): add output_descs for apply_on_physical_tensor
GitOrigin-RevId: 5b036c2c5a
|
3 years ago |
Megvii Engine Team
|
a5af35c18c
|
refactor(imperative): remove command buffer
GitOrigin-RevId: 83c8cb6d3b
|
3 years ago |
Megvii Engine Team
|
de8ffe0c12
|
refactor(imperative): unify interpreter option setting
GitOrigin-RevId: 53510445cc
|
3 years ago |
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
|
4 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 |