Megvii Engine Team
fefd4f0928
fix(traced_module): fix tensor node init
GitOrigin-RevId: bdd8259ce7
3 years ago
Megvii Engine Team
11d75fecb5
feat(dnn/check_non_finite): add batch check_non_finite
GitOrigin-RevId: e108133282
3 years ago
Megvii Engine Team
7a023c059a
feat(mge/traced_module): add optimization api
GitOrigin-RevId: eaa7402640
3 years ago
Megvii Engine Team
6c692b26a7
feat(mge/traced_module): add some fuse passes
GitOrigin-RevId: 065f9df32e
3 years ago
Megvii Engine Team
b28ad4e884
feat(mge/traced_module): add pattern match for TracedModule
GitOrigin-RevId: 0af7b076e6
3 years ago
Megvii Engine Team
ea3eb66e7f
docs(mge): modify some docstring about conv and tools
GitOrigin-RevId: 4833e78d71
3 years ago
Megvii Engine Team
acd835d0df
docs(mge): add some docs for cvt_color and GraphInference
GitOrigin-RevId: d78628bfcf
3 years ago
Megvii Engine Team
0185c1a9b9
feat(mge/traced_module): add argspec for top TracedModule
GitOrigin-RevId: 8e31a00c7e
3 years ago
Megvii Engine Team
1daeba76fd
fix(mge/traced_module): fix bn opdef loader
GitOrigin-RevId: d73f953ef6
3 years ago
Megvii Engine Team
2d20f93777
fix(mge/traced_module): fix merge GetAttr failure when qualname is incorrect
GitOrigin-RevId: 46241e5b4f
3 years ago
Megvii Engine Team
e6c271ae46
fix(mge/traced_module): fix some bugs for graph surgery
GitOrigin-RevId: 6328a84cbc
3 years ago
Megvii Engine Team
66f1358666
fix(mge/network): add warning when the node is not in the output list
GitOrigin-RevId: 3240546139
3 years ago
Megvii Engine Team
7b19bc76fb
feat(traced_module): support traced module backward compatible serialization
GitOrigin-RevId: aaa9e51c74
3 years ago
Megvii Engine Team
05ee603843
fix(mge/distributed): fix single_machine_cache error
GitOrigin-RevId: be57111408
3 years ago
Megvii Engine Team
0a56a5b110
feat(cuda/comp_node): gets (maximum) GPU memory allocated/reserved
GitOrigin-RevId: da2cc22436
3 years ago
Megvii Engine Team
5c94db16f2
feat(mge/functional): add groups support for conv_transpose2d & 3d
GitOrigin-RevId: b75b792fb4
3 years ago
Megvii Engine Team
f2f3356572
fix(mge): fix some minor problems
GitOrigin-RevId: 43abda2ab9
3 years ago
Megvii Engine Team
8a222d2c9c
fix(mge): replace functional paramter
GitOrigin-RevId: af3c68c588
3 years ago
Megvii Engine Team
27aa648ba7
fix(mge): control compute_mode by context
GitOrigin-RevId: f3968309f4
3 years ago
Megvii Engine Team
38b7cfdec1
fix(mge/utils): fix module states input is dict or others
GitOrigin-RevId: f9701b6134
3 years ago
Megvii Engine Team
846cc13ca8
fix(mge/functional): fix default dtype of F.full
GitOrigin-RevId: 5cd34af9bf
3 years ago
MegChai
63bb7c2946
docs(mge/functional): update functional.arange docstring
From GITHUB 386
COPYBARA_INTEGRATE_REVIEW=https://github.com/MegEngine/MegEngine/pull/386 from MegChai:docstring-arange 6702c6a5cd
Closes #386
GitOrigin-RevId: e27cdaefe8
3 years ago
Megvii Engine Team
a2a09ef990
fix(imperative): release dtr related resources when disable dtr
GitOrigin-RevId: eacfded9de
3 years ago
Megvii Engine Team
2676fb7344
refactor(imperative): remove enable_defrag interface as it is enabled by default
GitOrigin-RevId: de9e7d7f16
3 years ago
Megvii Engine Team
d909950f64
refactor(imperative): remove swap in dtr
GitOrigin-RevId: 5c9e42f74a
3 years ago
Megvii Engine Team
829f0907e7
fix(mge/traced_module): fix insert qat module
GitOrigin-RevId: 35849bc1a2
3 years ago
Megvii Engine Team
e02f1e07de
fix(ci): replace old lar with new lar
GitOrigin-RevId: e4117ff53b
3 years ago
Megvii Engine Team
edfd38befd
fix(mge/traced_module): fix node naming in the flattened graph
GitOrigin-RevId: aa7c516725
3 years ago
Megvii Engine Team
b877657456
ci(pytest): fixture doctest namespace in pytest config
GitOrigin-RevId: 90ef1924dc
3 years ago
Megvii Engine Team
ba8bd01023
fix(mge/traced_module): fix insert module
GitOrigin-RevId: 755e1c68f6
3 years ago
Megvii Engine Team
b8316de5a5
fix(mge/traced_module): patch missed functional in functional.nn
GitOrigin-RevId: 5aa1316fb1
3 years ago
Megvii Engine Team
dbca32702a
fix(mge/traced_module): fix module dict
GitOrigin-RevId: d7baf00e3c
3 years ago
Megvii Engine Team
01d2473ce7
fix(mge/traced_module): fix TracedModule flatten
GitOrigin-RevId: 7b15fe492b
3 years ago
Megvii Engine Team
a404c508e9
feat(mge): support dump with specific format
GitOrigin-RevId: 57a7c0de02
3 years ago
Megvii Engine Team
fba523a196
fix(mge/device): fix the directive content block
GitOrigin-RevId: 539feb7956
3 years ago
Megvii Engine Team
25ec2530ba
feat(whl/api/lar): enable megengine dll on Windows
1: reduce python whl package size
2: unify api link logic on all OS
3: add option: MGE_WINDOWS_BUILD_WITH_STATIC_CRT
--- default OFF
--- if build CRT(vc runtime) with STATIC with megengine.dll
some CRT api will crash, for example, flush, so if you
build with static megengine, and do not want to install CRT
you can set MGE_WINDOWS_BUILD_WITH_STATIC_CRT TRUE
--- how to install CRT:
https://docs.microsoft.com/en-us/cpp/windows/latest-supported-vc-redist?view=msvc-160
install VC_redist.x64.exe
4: rename megengine_export to megengine_shared(only export needed symbols ),
caused by runtime symbols conflict with pytorch
GitOrigin-RevId: 93d8d80f29
3 years ago
Megvii Engine Team
feea43bc74
fix(mge/traced_module): associate name with node
GitOrigin-RevId: 8d9a59bade
3 years ago
Megvii Engine Team
a6fe7f7ff2
refactor(mge/traced_module): refactor Node naming rule and merge GetAttr
GitOrigin-RevId: a43ad1273c
3 years ago
Megvii Engine Team
ac86d64474
fix(mge/module): fix prelu error when use_symbolic_shape is true
GitOrigin-RevId: 25b9c4d41d
3 years ago
Megvii Engine Team
4f6c5d8f01
feat(mge/dump): enable jit.dump to dump with testcase
GitOrigin-RevId: 5dce356452
3 years ago
Megvii Engine Team
182ca25d45
feat(mge/imperative): add coalesce_all_free_memory mge api
GitOrigin-RevId: ea05b1e9a1
3 years ago
Megvii Engine Team
9d05738d54
Merge pull request #225 from chenjiahui0131:pixel_shuffle_module
GitOrigin-RevId: d94533655a
3 years ago
Megvii Engine Team
936bb23738
fix(mge/tensor): fix is/is not SyntaxWarning occurs in python3.8
GitOrigin-RevId: 12b66da367
3 years ago
Megvii Engine Team
8d507cc39d
docs(mge/module): add note about definition of momentum in BatchNorm
GitOrigin-RevId: 30a9aec20f
3 years ago
Megvii Engine Team
2c5a007a92
fix(mge/optimizer): allow lr to be 0
GitOrigin-RevId: dabd1fcc33
3 years ago
chenjiahui
c8a9a3fd0a
feat(imperative): add PixelShuffle to module.__init__.py
3 years ago
Megvii Engine Team
29f9935d92
fix(imperative/python): add layer_norm doc and rm useless param
GitOrigin-RevId: 1b15db621e
3 years ago
Megvii Engine Team
300955140e
fix(traced_module/doc): fix warning for traced_module docstring
GitOrigin-RevId: 5d2d047d2f
3 years ago
Megvii Engine Team
f7d05db7d6
fix(mge/module): fix named_tensors
GitOrigin-RevId: bb5aa1f41d
3 years ago
Megvii Engine Team
2108696588
feat(mge/imperative): add local response normalization
GitOrigin-RevId: 939a4d26dd
3 years ago