Megvii Engine Team
fb20cb36a0
docs(mge/traced_module): update traced_module api doc
GitOrigin-RevId: 19a95d26c7
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
763c56f3b9
feat(imperative): add traced module
GitOrigin-RevId: 28c3503f2e
4 years ago
Megvii Engine Team
1bf18252c4
feat(mge/amp): add mix precision autocast support
GitOrigin-RevId: 6fbffc4845
4 years ago
Megvii Engine Team
90d3905799
feat(mge): add mge._exit
GitOrigin-RevId: e7f348a677
4 years ago
Megvii Engine Team
3352fe7003
refactor(dtr): import dtr as submodule
GitOrigin-RevId: abecd0f176
4 years ago
Megvii Engine Team
601a33a84d
refactor(mge/dtr): update dtr api
GitOrigin-RevId: dc366c65be
4 years ago
Megvii Engine Team
7af1ec66db
fix(tensor): del valid tensors when compnode finalizing
GitOrigin-RevId: bace1f2b51
4 years ago
Megvii Engine Team
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
4 years ago
Megvii Engine Team
08cc10324e
fix(imperative): fix persistent_cache
GitOrigin-RevId: 8f7bb5899f
4 years ago
Megvii Engine Team
09af925fb6
fix(mge): fix cpp trace function release
GitOrigin-RevId: 924f945c21
4 years ago
Megvii Engine Team
6919127fd4
fix(mge): fix event pool destructor order issue and do full sync at python exit
GitOrigin-RevId: b9fa1f45b1
4 years ago
Megvii Engine Team
60459bd79b
fix(mge/pytest): remove sync in pytest teardown & add full_sync
GitOrigin-RevId: ee278a8ac1
4 years ago
Megvii Engine Team
89b6dbc74c
fix(mge/imperative): import subpackages
GitOrigin-RevId: 342f6c87cc
4 years ago
Megvii Engine Team
4d75f691a0
feat(mge): restore remote send/recv
GitOrigin-RevId: 8b78fd5591
4 years ago
Megvii Engine Team
243a05b410
fix(mge): fix cpp trace function release
GitOrigin-RevId: 73f9642821
4 years ago
Megvii Engine Team
b5e46ae92f
feat(mge): restore Function
GitOrigin-RevId: dd455238ba
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
ab6328c563
feat(imperative): port persistent cache
GitOrigin-RevId: 8ca24a37cc
4 years ago
Megvii Engine Team
8118a5946d
fix(mge/utils): fix get_oprs_seq of cgtools
GitOrigin-RevId: 366a56f4d5
4 years ago
Megvii Engine Team
5cc043f015
fix(imperative): sync command queue at exit
GitOrigin-RevId: c0d6aad275
4 years ago
Megvii Engine Team
92afcf3eb4
refactor(mge/imperative): move cgtools to megengine.utils and add load_and_inference in cgtools
GitOrigin-RevId: abfee3d4fa
4 years ago
Megvii Engine Team
ac11c38a97
feat(mge/imperative): add graph load and cgtools for imperative
GitOrigin-RevId: ba251f452a
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
6e882c1a86
feat(whl/imperative): compat for build python whl imperative and legacy runtime
GitOrigin-RevId: 7f6629ae1f
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