Megvii Engine Team
865fbe16c0
feat(imperative/utils): add optimize-for-inference interface for opgraph
GitOrigin-RevId: 9f93f82190
4 years ago
Megvii Engine Team
c92317edc0
fix(mge/utils): filter out parameter "arg_names" and "output_name" in network.dump
GitOrigin-RevId: 408f52ad2d
4 years ago
Megvii Engine Team
24b91b98c7
feat(mge/utils): add array method for varnode
GitOrigin-RevId: 6e4d05b475
4 years ago
Megvii Engine Team
5e54fdc32b
fix(imperative/utils): fix name filter of network varnode
GitOrigin-RevId: 20ead35335
4 years ago
Megvii Engine Team
7c715bd42e
fix(mge/utils): fix module stats calculate flops bug for group conv and remove model status change
GitOrigin-RevId: 647dc6eb66
4 years ago
Megvii Engine Team
245a3f8129
fix(mge/tools): fix module stats' receptive field bug for Module
GitOrigin-RevId: b471363830
4 years ago
Megvii Engine Team
007a2376c3
fix(mge/tools): fix node display bug in tensorboard
GitOrigin-RevId: c997d6cccb
4 years ago
Megvii Engine Team
b10238acd1
feat(mge/tools): add support of receptive_field stats for NetworkNode
GitOrigin-RevId: 11ef335468
4 years ago
Megvii Engine Team
84c2a5c27a
feat(mge/tools): add summary print for module_stats and network_visualize
GitOrigin-RevId: 7d85aa0ea2
4 years ago
Megvii Engine Team
e9982d613f
chore(mge/functional): add compatible code for functional api
GitOrigin-RevId: 3b2f829cc5
4 years ago
Megvii Engine Team
c45f1eb298
fix(mge/tools): improve `module_visualize` result's robustness and beauty
GitOrigin-RevId: ef7b573776
4 years ago
Megvii Engine Team
786c36ff42
fix(mge/tools): rename `net_stats` in function and examples to match file name
GitOrigin-RevId: 82a1377d66
4 years ago
Megvii Engine Team
c8697a7005
feat(imperative/src): python wrapper for cambricon and atlas runtime opr
GitOrigin-RevId: bd969d1339
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
fad5bc74c3
feat(mge/tools): add load_network_and_run for megengine
GitOrigin-RevId: 7d8ed5af79
4 years ago
Megvii Engine Team
813be72ce9
feat(mge/utils): refactor GraphInference and add more options
GitOrigin-RevId: 44b96dbf3d
4 years ago
Megvii Engine Team
d970b85deb
feat(functional): add conv3d
GitOrigin-RevId: c06ac77c6c
4 years ago
Megvii Engine Team
53075cd3da
feat(mge/experimental): add visualization and net stats for python graph
GitOrigin-RevId: a1ab77c20a
4 years ago
Megvii Engine Team
ae3123b315
feat(mge): add python graph for mgb graph editing
GitOrigin-RevId: 6a9d5beba2
4 years ago
Megvii Engine Team
6fb19b664b
feat(imperative/src): name operators automatically when tracing
GitOrigin-RevId: ff8eb003c5
4 years ago
Megvii Engine Team
91a3580f75
refactor(mge/cgtools): remove load_and_inference and use GraphInference
GitOrigin-RevId: 0e688ebd59
4 years ago
Megvii Engine Team
334eda8717
refactor(mge): test trace inside opr_test
GitOrigin-RevId: 2cf1135c1c
4 years ago
Megvii Engine Team
dbb3dd681f
refactor(profiler): integrate profiler into interpreter
GitOrigin-RevId: ccc984acbd
4 years ago
Megvii Engine Team
8e50a6daa7
feat(mge/utils): add GraphInference in cgtools
GitOrigin-RevId: 72f2201169
4 years ago
Megvii Engine Team
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
4 years ago
Megvii Engine Team
ca1d74ad3f
fix(mge/utils): fix types import error
GitOrigin-RevId: d2d4bd2725
4 years ago
Megvii Engine Team
75129cf071
chore(mge): clean up before merge to dev
* remove dead test
* clean some codes
* fix test_fake_quant due to change of implementation
GitOrigin-RevId: f030a9966d
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
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
88b402ef62
feat(mge/trace): tracing use id to set priority
GitOrigin-RevId: 6e1f1ece0e
4 years ago
Megvii Engine Team
69e3e32240
feat(imperative): auto generated opdef header and python binding
GitOrigin-RevId: d2f22ad5fe
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
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
b309890c66
docs(mge): pytest for sphinx docstring
GitOrigin-RevId: 8bed12562a
4 years ago
Megvii Engine Team
20cdd11d4d
docs(mge): fix docstring of cgtools
GitOrigin-RevId: ee8c93b9f7
4 years ago
Megvii Engine Team
2c9fa7f650
feat(mge/profiler): add compatible and graphviz mode
GitOrigin-RevId: 40ca9ea80e
4 years ago
Megvii Engine Team
026af62042
docs(mge): docs typo fix
GitOrigin-RevId: 851f6de02f
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
b7e9716098
feat(mge/plugin): add load_binary_tensor, copybara add comp_binary_iodump
GitOrigin-RevId: 8bb5d439d1
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
6f581906a1
refactor(mge/profiler): refactor profiler
GitOrigin-RevId: 279aa779a6
4 years ago
Megvii Engine Team
e9104ef157
fix(mge/parampack): fix copy stream, import cycle
GitOrigin-RevId: 673e11c5b6
4 years ago
Megvii Engine Team
b1ab3646f5
feat(imperative): add tensor sanity check
GitOrigin-RevId: 27243978e3
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