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
Megvii Engine Team
7234efe1e7
feat(opr): let random op support empty output
GitOrigin-RevId: a2174975aa
3 years ago
Megvii Engine Team
59d5976622
fix(profiler): collect records when thread exit
GitOrigin-RevId: fb796dd61c
4 years ago
Megvii Engine Team
dff7719e87
feat(mge/distributed): add preload host data with op
fix(mge/distributed): change api name with preload
fix(mge/distributed): fix recursive model in preload tensor
fix(mge/distributed): fix recursive when cache contain None
GitOrigin-RevId: 80e2a6dd70
3 years ago
Megvii Engine Team
4688bbabad
refactor(functional): hide functional.vision and replace with functional.nn
GitOrigin-RevId: b6888c7f65
3 years ago
Megvii Engine Team
8796586b89
refactor(functional): import all from metric in nn
GitOrigin-RevId: 41ab78d78d
3 years ago
Megvii Engine Team
27f2ecc436
feat(imperative/opr): add extern c opr and support dump
GitOrigin-RevId: b29e374c60
3 years ago
Megvii Engine Team
91264f3797
fix(traced_module): fix __getattr__ of TracedModuleBuilder
GitOrigin-RevId: 94d91d6938
3 years ago
Megvii Engine Team
526c82c858
feat(traced_module): delete value of node when it will not be used by any expr
GitOrigin-RevId: 3fb7350d01
3 years ago
Megvii Engine Team
6ce1483bc9
fix(utils/network): fix replace var in different networks
GitOrigin-RevId: 8d1356ddb1
3 years ago
Megvii Engine Team
1019865071
feat(module): add tensors and named_tensors
GitOrigin-RevId: cb56d65d38
3 years ago
Megvii Engine Team
bd817f3af2
perf(syncbn): fallback to bn when sync is not required
GitOrigin-RevId: e89fdd87b0
3 years ago
Megvii Engine Team
ffcb4dacd9
style(mge): change the pad opr doc and format
GitOrigin-RevId: 7b20e96519
3 years ago