Megvii Engine Team
cad8568c34
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
Megvii Engine Team
9748aebeea
refactor(mge): tensor_shape -> symbolic_shape
GitOrigin-RevId: 366dc048bf
4 years ago
Megvii Engine Team
f4b169322d
test(mgb/imperative): add adaptive pooling pytest
GitOrigin-RevId: c4dfed1f80
4 years ago
Megvii Engine Team
8d1f302590
fix(mge): fix batch norm dump
GitOrigin-RevId: eb739437ef
4 years ago
Megvii Engine Team
7d8a7e3efe
feat(mge): do not export F.loss.*
GitOrigin-RevId: a3ce4d3d76
4 years ago
Megvii Engine Team
5b2f0129cb
feat(mge): rename cross_entropy_with_softmax -> cross_entropy
GitOrigin-RevId: 9435c3260a
4 years ago
Megvii Engine Team
6ce796aec4
test(mge): replace `assertTensorClose` with `np.testing.assert_allclose`
GitOrigin-RevId: 2bf8c42cfd
4 years ago
Megvii Engine Team
10e942d952
refactor(mge): polish api
- refactor(mge): add support for optimizer.step().clear_grad() idiom
- refactor(mge): rename some methods of GradManager
- refactor(mge): remove tensor_nn and TensorDict
- refactor(mge): remove Buffer
- refactor(mge): remove requires_grad flag
- refactor(mge): add a default grad=None attribute to Tensor
- refactor(mge): deprecation for 1.0
GitOrigin-RevId: 3b723d9387
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
e283663a02
fix(mge/imperative): update tests to new optimizer api
GitOrigin-RevId: 3d06e3db3c
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
087ceb52ef
feat(mge/imperative): add more optimizer trace tests
GitOrigin-RevId: 4127de1d22
4 years ago
Megvii Engine Team
0e82b959a1
feat(mge/imperative): add sublinear options
GitOrigin-RevId: f0e917f716
4 years ago
Megvii Engine Team
afddefb677
feat(mge/imperative): add more trace test
GitOrigin-RevId: b02e420a8a
4 years ago
Megvii Engine Team
e1fba6ece7
test(mge/distributed): add get_device_count_by_fork to fix distributed test skip
GitOrigin-RevId: 9ffd8a6149
4 years ago
Megvii Engine Team
6e882c1a86
feat(whl/imperative): compat for build python whl imperative and legacy runtime
GitOrigin-RevId: 7f6629ae1f
4 years ago
Megvii Engine Team
40d18c8920
fix(mge/imperative): fix tests when shape is tensor
GitOrigin-RevId: fd0095c1ec
4 years ago
Megvii Engine Team
227ec04bd3
fix(mge/test): fix pytest failed issue
GitOrigin-RevId: 96df470406
4 years ago
Megvii Engine Team
6b380e8965
feat(mge/imperative): run oss test and restore cmake list build items
GitOrigin-RevId: 11411b6964
4 years ago