Megvii Engine Team
d90cb7763c
feat(src/core): record support change ptr basic
GitOrigin-RevId: b12f1c4a66
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
ffbfe59c9f
fix(mge/subgraph): fix mem leak
GitOrigin-RevId: 48beb625e1
3 years ago
Megvii Engine Team
89186edc5d
fix(dnn): correct reduce/argmxx/fakequant calculation with nan
GitOrigin-RevId: 7e78bdae91
3 years ago
Megvii Engine Team
68cdabd288
feat(opr): indexing_multi_axis_vec support nd index
GitOrigin-RevId: 07b1248bdc
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
7591718d20
feat(mge): add functional test
GitOrigin-RevId: aa0be62686
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
1613135990
fix(imperative): insert to candidate only if pinned is 0
GitOrigin-RevId: 2b849d7fe2
3 years ago
Megvii Engine Team
846cc13ca8
fix(mge/functional): fix default dtype of F.full
GitOrigin-RevId: 5cd34af9bf
3 years ago
Megvii Engine Team
0ddabb06cc
fix(imperative/astype): fix astype when target dtype is int4
GitOrigin-RevId: 1c5195176d
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
89ed7ab2ff
test(imperative): speed up dtr test
GitOrigin-RevId: 57f092d729
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
bcf69d8f87
refactor(imperative): correctly apply sqrt sampling for dtr
GitOrigin-RevId: dabd365517
3 years ago
Megvii Engine Team
481007812e
fix(mgb): support rc version
GitOrigin-RevId: aa5fe5198d
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
23c1fda7e6
perf(arm_common): optimize sigmoid
GitOrigin-RevId: 7cb248a15b
3 years ago
Megvii Engine Team
7fa5f6f4e2
fix(mgb): correct version of fbs serialization
GitOrigin-RevId: 2a71d9afc2
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
669c3cda85
feat(mge): enable defrag by default
GitOrigin-RevId: 70bd4bb119
3 years ago
Megvii Engine Team
d9a9d9d49e
fix(imperative/fastrun): set workspace limit for imperative rt
GitOrigin-RevId: 474dc691a3
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
676b205bd8
style(imperative): add bypass of clang-format for cg options
GitOrigin-RevId: 824d610631
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
fe15239ac0
fix(imperative): fix error message for tensors with intbx data type
GitOrigin-RevId: cbb42f8127
3 years ago