Megvii Engine Team
7ac693a882
docs(imperative): fix multiple target warning when cross-reference
GitOrigin-RevId: c5b4884c8a
3 years ago
Megvii Engine Team
f2f3356572
fix(mge): fix some minor problems
GitOrigin-RevId: 43abda2ab9
3 years ago
Megvii Engine Team
c5f8b58385
docs(docstring): transfer to google style
GitOrigin-RevId: a71245c553
3 years ago
Megvii Engine Team
3e58cbb8d2
fix(mge/quantization): fix lsq init format error
GitOrigin-RevId: 032de9c773
3 years ago
Megvii Engine Team
3eb0505f9b
feat(imperative): add support for quantized conv transpose2d
GitOrigin-RevId: ffd6431299
3 years ago
Megvii Engine Team
bfc4e7a966
docs(mge): fix amp docstring problems
GitOrigin-RevId: e5540eb940
3 years ago
liuke
0baf6b0d63
Merge pull request #175 from tpoisonooo:fix-spell-error
GitOrigin-RevId: fe649676ac
3 years ago
Megvii Engine Team
1bf18252c4
feat(mge/amp): add mix precision autocast support
GitOrigin-RevId: 6fbffc4845
4 years ago
Megvii Engine Team
f30c0e06a6
feat(mgb/opr): add lsq opr
GitOrigin-RevId: 45494a2b57
4 years ago
tpoisonooo
7038a7f5d0
fix(quant): fix spell error
4 years ago
Megvii Engine Team
05550bc5b1
fix(mge/quantization): fix a compute problem
GitOrigin-RevId: 8aaedea959
4 years ago
Megvii Engine Team
44bafd3f58
fix(imperative/quantization): fix zero scale bug of easy quant
GitOrigin-RevId: f45e19b3e4
4 years ago
Megvii Engine Team
dc96f6aa49
fix(mge/quantization): fix quantized concat forward problem
GitOrigin-RevId: dc21b340d1
4 years ago
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