Megvii Engine Team
992a90bbad
docs(mge/quantization): add docstring for Observer
GitOrigin-RevId: 043be3886d
4 years ago
Megvii Engine Team
9c90ce8c33
fix(mge/quantization): fix `quantize` and `quantize_qat`'s `set_expand_structure` arguments
GitOrigin-RevId: c61633095d
4 years ago
Megvii Engine Team
5a38ad3920
feat(mge/utils): add get/set_expand_structure to deal with complex key
GitOrigin-RevId: 4d1b952068
4 years ago
Megvii Engine Team
1670a8cf96
docs(mge/quantization): fix wrong unexpected unindent in docstring
GitOrigin-RevId: e604bcebb2
4 years ago
Megvii Engine Team
ce88e6c4f7
feat(mge/quantization): use extra act_fakequant to decide whether to do bias fakequant
GitOrigin-RevId: bf54012155
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
4130dcd355
fix(mge/quantization): fix QATModule filter in `reset_qconfig` and `hook_qat_module`
GitOrigin-RevId: 92e9f36ca4
4 years ago
Megvii Engine Team
5b697c714c
chore(imperative): remove unused reference to megbrain_graph
GitOrigin-RevId: 2c31f13bb8
4 years ago
Megvii Engine Team
d1be31279a
fix(mge/quantization): fix tqt load and convert issue and observer calculate params issue
GitOrigin-RevId: f8511f72ad
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
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
4 years ago
Megvii Engine Team
5100317660
feat(mge): make Function available at autodiff.Function
GitOrigin-RevId: 7c9ab65351
4 years ago
Megvii Engine Team
b7ed0cb850
fix(mge/quantization): replace `_reset` with "=" in observer
GitOrigin-RevId: ed6af9b98d
4 years ago
Megvii Engine Team
a85531dd0f
feat(mgb/opr): add tqt opr
GitOrigin-RevId: 49c62cd532
4 years ago
Megvii Engine Team
60c7d62a4c
refactor(imperative): remove multidispatch, raw_tensor, register
GitOrigin-RevId: ca5a6ed8eb
4 years ago
Megvii Engine Team
b5e46ae92f
feat(mge): restore Function
GitOrigin-RevId: dd455238ba
4 years ago
Megvii Engine Team
94796060b6
test(mge/quantization): qat module, quantized op and module
GitOrigin-RevId: 17b28060cc
4 years ago
Megvii Engine Team
ab9f44f15c
feat(mge/quantization): add support for easyquant
GitOrigin-RevId: 060d908349
4 years ago
Megvii Engine Team
6c4841e807
fix(mge/quantization): `disable_fake_quant` does not work correctly
GitOrigin-RevId: 0c568d3335
4 years ago
Megvii Engine Team
638ab52fdc
feat(mge/imperative): simulates scalar
GitOrigin-RevId: e81630e256
4 years ago
Megvii Engine Team
33e8879af4
feat(mge/quantization): support distributed qat
GitOrigin-RevId: c915c843b8
4 years ago
Megvii Engine Team
ae8c3c8163
feat(mge/functional): add python wrapper for fake quant opr
GitOrigin-RevId: 5f205198be
4 years ago
Megvii Engine Team
b309890c66
docs(mge): pytest for sphinx docstring
GitOrigin-RevId: 8bed12562a
4 years ago
Megvii Engine Team
d8ac6c70a1
fix(quantize): fix quantize calibration dtype issue
GitOrigin-RevId: 667c99f469
4 years ago
Megvii Engine Team
205a39b07d
feat(mge): remove add_update
GitOrigin-RevId: 3068593eeb
4 years ago
Megvii Engine Team
026af62042
docs(mge): docs typo fix
GitOrigin-RevId: 851f6de02f
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
b3889938dc
feat(mge/examples): add trace & dump example of cifar10 quantization
GitOrigin-RevId: cfc5e3483a
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