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
Megvii Engine Team
ca4c93dee7
fix(mge): fix layer norm amp bug
GitOrigin-RevId: dba691fcbf
3 years ago
Megvii Engine Team
cebda6ff8d
feat(mge/imperative): add ctc loss
GitOrigin-RevId: e29854a98e
3 years ago
Megvii Engine Team
f5cb21ed3a
fix(mgb/opr): add non finite check
GitOrigin-RevId: a9fcd0a350
3 years ago
Megvii Engine Team
4e95c13617
feat(sgd): sgd supports nesterov momentum
GitOrigin-RevId: 13eda179da
3 years ago
Megvii Engine Team
c0ccd0ea7e
feat(mge/bn): add NHWC support for bn
GitOrigin-RevId: 0a5bb6f72d
3 years ago
chenjiahui
6b863cc53a
feat(imperative): add pixel_shuffle opr
Closes #211
GitOrigin-RevId: 1846714791
3 years ago
Megvii Engine Team
b982be567a
feat(mge/imperative): add permutation support for the tensor
GitOrigin-RevId: 7ed0447bfe
3 years ago
Megvii Engine Team
3977b7aa0b
feat(mgb/shuffle): add shuffle opr
GitOrigin-RevId: 80490a6f84
3 years ago
Megvii Engine Team
11a383bc79
feat(mgb): add tensorboard tool c++ layer interface and python parsing
GitOrigin-RevId: ae98bc887f
3 years ago
Megvii Engine Team
8084e4e2ee
feat(mgb): add tensorboard tool python layer interface
GitOrigin-RevId: 065bc4d153
3 years ago
Megvii Engine Team
97b1b7774d
feat(mgb): add megbrain layer norm opr with subgraph
GitOrigin-RevId: 9b7fa821f8
3 years ago
Megvii Engine Team
65d554ed8e
refactor(dtr): improve dtr in computing graph
GitOrigin-RevId: d599a08e52
3 years ago
Megvii Engine Team
f642b05e17
test(mge): update traced_module unit test
GitOrigin-RevId: 3948d50d79
3 years ago
Megvii Engine Team
fb20cb36a0
docs(mge/traced_module): update traced_module api doc
GitOrigin-RevId: 19a95d26c7
3 years ago
Megvii Engine Team
c7a8d945c7
fix(mge/traced_module): let graph record total_id
GitOrigin-RevId: f99178f3ac
3 years ago
Megvii Engine Team
4de62ad6cd
docs(mge/device): fix docstring of get_cuda_compute_capability
GitOrigin-RevId: 4ec04e8b9b
3 years ago
Megvii Engine Team
cdb692d2fa
refactor(imperative): add TODO tag for some functions
GitOrigin-RevId: e295a1fa55
3 years ago
Megvii Engine Team
90dd07161c
refactor(imperative): modify the python interface of custom op
GitOrigin-RevId: e82e5de480
3 years ago
Megvii Engine Team
cbf024bf20
feat(imperative): add adaptor between custom op and imperative runtime
GitOrigin-RevId: d7877f2e32
3 years ago
Megvii Engine Team
43fb2c4afb
feat(opr): let roll support empty IO
GitOrigin-RevId: b9a59b623a
3 years ago