Megvii Engine Team
c5f8b58385
docs(docstring): transfer to google style
GitOrigin-RevId: a71245c553
3 years ago
Megvii Engine Team
bfc4e7a966
docs(mge): fix amp docstring problems
GitOrigin-RevId: e5540eb940
3 years ago
Megvii Engine Team
1bf18252c4
feat(mge/amp): add mix precision autocast support
GitOrigin-RevId: 6fbffc4845
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
8be70d170c
docs(misc): correct docstring format broadly
GitOrigin-RevId: 45234ca07e
4 years ago
Megvii Engine Team
aed681d325
feat(imperative/utils): optimize the naming rules
GitOrigin-RevId: 329bac640a
4 years ago
Megvii Engine Team
84f990a051
fix(mge/tensor): fix `__repr__` for Tensor's subclass
GitOrigin-RevId: 91b58e6dca
4 years ago
Megvii Engine Team
cdbb4a20ff
fix(mge/tensor): fix tensor's serialization behavior
GitOrigin-RevId: 4d74a4b46e
4 years ago
Megvii Engine Team
e6b06914f7
feat(mge/tensor): support non-Tensor value in `_reset` and remove depreciated tests
GitOrigin-RevId: faf6c78aa8
4 years ago
Megvii Engine Team
fa4bf16800
feat(mge/functional): add repeat and tile opr
GitOrigin-RevId: a20d4b6fb0
4 years ago
Megvii Engine Team
1d7dd00144
feat(mge/quantization): add QParams and QuantDtypeMeta for quantization data structure
GitOrigin-RevId: df3416fe13
4 years ago
Megvii Engine Team
d0f70a445d
fix(imperative/utils): module parameters' name do not have scope after dumping
GitOrigin-RevId: 1497272294
4 years ago
Megvii Engine Team
4f3875eb4f
docs(mge): add missing docstring and fix sphinx build warnings
GitOrigin-RevId: 4ce73cfd80
4 years ago
Megvii Engine Team
6fb19b664b
feat(imperative/src): name operators automatically when tracing
GitOrigin-RevId: ff8eb003c5
4 years ago
Megvii Engine Team
02df634da2
fix(mge/quantization): set ``q_dict`` as an instance property
GitOrigin-RevId: 2f32008aad
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
412d1f0cdc
fix(mge/tensor): implement abtract method to fix lint errors
GitOrigin-RevId: d53f2eac6a
4 years ago
Megvii Engine Team
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
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
dd9f54cdfa
refactor(mge): migrate to new core implementation
* remove dispatcher/interpreter python wrapper
* rename tensor_wrapper to array_method
GitOrigin-RevId: b8a402c2be
4 years ago
Megvii Engine Team
87f4b46e3f
perf(mge/imperative): move convert_inputs from python to C++
GitOrigin-RevId: baef3d348c
4 years ago
Megvii Engine Team
b310f2615b
feat(mge/imperative): implement trace and dump under new core implementation
GitOrigin-RevId: 4edc38eaf2
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
34c705fcbf
refactor(mge/imperative): move detach into C++
GitOrigin-RevId: 8c0d86cbbf
4 years ago
Megvii Engine Team
147cef52a0
refactor(mge/imperative): implement new tensor system
GitOrigin-RevId: 2dd4e460ac
4 years ago
Megvii Engine Team
b309890c66
docs(mge): pytest for sphinx docstring
GitOrigin-RevId: 8bed12562a
4 years ago
Megvii Engine Team
2627e1f7df
fix(mge/grad_manager): allow multiple calls of `release`
GitOrigin-RevId: 38ca4c78ff
4 years ago
Megvii Engine Team
2e9ba679b5
feat(mge/device): __repr__ method will show physical device
GitOrigin-RevId: 050c3864a7
4 years ago
Megvii Engine Team
d225cbcdbe
feat(mge): add device name check
GitOrigin-RevId: d9910b6275
4 years ago
Megvii Engine Team
4ab259f558
refactor(mge): add some deprecation and migration guides
GitOrigin-RevId: 4793adf02b
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
44d0b5daf5
feat(imperative): enable to() to copy to device
GitOrigin-RevId: f9caf17d24
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