Megvii Engine Team
a51d5b4c31
feat(mgb/jit): add mlir backend for cpu and cuda
GitOrigin-RevId: 814fed047e
4 years ago
Megvii Engine Team
523ce65e0d
fix(mge/imperative): fix cgtools related tests
GitOrigin-RevId: 8f1eadb32e
4 years ago
Megvii Engine Team
dd39265e95
fix(mgb/dtype): enable TypeCvt for bool when trace(symbolic=True)
GitOrigin-RevId: 4e0fc63369
4 years ago
Megvii Engine Team
5c37a64e3f
feat(sdk/load_and_run): add dump with testcase for mge support imperative
GitOrigin-RevId: 17a0cf6293
4 years ago
Megvii Engine Team
ac11c38a97
feat(mge/imperative): add graph load and cgtools for imperative
GitOrigin-RevId: ba251f452a
4 years ago
Megvii Engine Team
76f367962f
fix(mge/trace): fix op order in symbolic
GitOrigin-RevId: fbf081a199
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
9389a805d8
fix(mge/oprmm): fix grad for collective comm
GitOrigin-RevId: e55dc47a6e
4 years ago
Megvii Engine Team
6f581906a1
refactor(mge/profiler): refactor profiler
GitOrigin-RevId: 279aa779a6
4 years ago
Megvii Engine Team
cc952b2b92
fix(rocm): fix rocm megdnntest sleep and a cut code
GitOrigin-RevId: 26de5ca98b
4 years ago
Megvii Engine Team
3a03fa7a50
fix(dnn/cuda): disable pascal sass conv2d
GitOrigin-RevId: 385d066595
4 years ago
Megvii Engine Team
5474b000a7
fix(mge/functional): fix convert_inputs before apply
GitOrigin-RevId: ab41974a1f
4 years ago
Megvii Engine Team
a5fad7d07c
feat(dnn): add compile for riscv64
GitOrigin-RevId: fa0c163527
4 years ago
Megvii Engine Team
478f2c51dc
feat(mge/jit): add trace/dump options
GitOrigin-RevId: 0f43c14599
4 years ago
Megvii Engine Team
44d0b5daf5
feat(imperative): enable to() to copy to device
GitOrigin-RevId: f9caf17d24
4 years ago
Megvii Engine Team
3e11d89415
fix(dnn/dump): add more info for dump CD4
GitOrigin-RevId: 5840afaacd
4 years ago
Megvii Engine Team
d49a5c1be2
fix(mge/imperative): remove codes newly introduced in numpy 1.19
GitOrigin-RevId: 874350cbc7
4 years ago
Megvii Engine Team
69a7c55f55
test(mge/function): fix test for new optimizer api
GitOrigin-RevId: 8ae7720fe6
4 years ago
Megvii Engine Team
a66d4b8bb8
fix(mge/parampacksplit): fix param pack split mem forward
GitOrigin-RevId: 8c001b73ff
4 years ago
Megvii Engine Team
bd3b9cb6d4
fix(mge/oprmm): fix grad for collective comm
GitOrigin-RevId: 8e28f46c90
4 years ago
Megvii Engine Team
8d02d10483
refactor(mge/distributed): change bcast_params_ to bcast_list_
GitOrigin-RevId: 26b452a6b7
4 years ago
Megvii Engine Team
c7acba41fc
refactor(mge/optimizer): refine gradmanager api, record = __enter__
GitOrigin-RevId: 5376177237
4 years ago
Megvii Engine Team
8c482b6709
fix(mge/grad): make register_after_backward_callback private
GitOrigin-RevId: 8eb6c0e628
4 years ago
Megvii Engine Team
66b6daf777
test(mge/optimizer): fix test for new optimizer api
GitOrigin-RevId: 482ee62652
4 years ago
Megvii Engine Team
e9104ef157
fix(mge/parampack): fix copy stream, import cycle
GitOrigin-RevId: 673e11c5b6
4 years ago
Megvii Engine Team
e283663a02
fix(mge/imperative): update tests to new optimizer api
GitOrigin-RevId: 3d06e3db3c
4 years ago
Megvii Engine Team
b5016b9d29
feat(mge/parampack): add parampack in allreduce callback
GitOrigin-RevId: 73d53eeba1
4 years ago
Megvii Engine Team
5ae89c799b
refactor(mgb/grad): place grad at param.grad
GitOrigin-RevId: fddeace402
4 years ago
Megvii Engine Team
9faa32fcb7
fix(mge/imperative): fix grad callback
GitOrigin-RevId: 6f843b0106
4 years ago
Megvii Engine Team
6d4fd9384d
fix(mge/optimizer): remove distributed optimizer
GitOrigin-RevId: 3e5d0612f0
4 years ago
Megvii Engine Team
e50fa074cb
fix(mge/imperative): remove backward from optimizer
GitOrigin-RevId: ad6ad444fa
4 years ago
Megvii Engine Team
6070266766
refactor(mge/grad_manager): refactor gradmanager, add allreduce callback
GitOrigin-RevId: 086e2871e8
4 years ago
Megvii Engine Team
3f2eac2fe1
fix(mge/imperative): move functional/distributed.py to distributed/functional.py
GitOrigin-RevId: 30cf2f514b
4 years ago
Megvii Engine Team
b3889938dc
feat(mge/examples): add trace & dump example of cifar10 quantization
GitOrigin-RevId: cfc5e3483a
4 years ago
Megvii Engine Team
67859f04e1
fix(imperative): add __array__ and __array_wrap__ for tensorwrapper
GitOrigin-RevId: 87d4ab6c8e
4 years ago
Megvii Engine Team
b1ab3646f5
feat(imperative): add tensor sanity check
GitOrigin-RevId: 27243978e3
4 years ago
Megvii Engine Team
5aa19f3d4a
test(mge/imperative): add more testcases for function
GitOrigin-RevId: e0675994f1
4 years ago
Megvii Engine Team
2a300415bc
fix(mge): remove a duplicated __init__ call in jit.trace
GitOrigin-RevId: d898f42fc4
4 years ago
Megvii Engine Team
517c1ec007
fix(mge): disable dedup for io callback oprs
GitOrigin-RevId: 424d347d83
4 years ago
Megvii Engine Team
939bf3b28d
fix(mge/imperative): impl hashable for nms and cond_take
GitOrigin-RevId: 56918db014
4 years ago
Megvii Engine Team
8dc23e0fdf
fix(mge/functional): fix indexing_one_hot and remote_recv
GitOrigin-RevId: 00bdfb502b
4 years ago
Megvii Engine Team
3bbfef3009
fix(mge/imperative): impl hashable for coll-comm
GitOrigin-RevId: 76ab16a89b
4 years ago
Megvii Engine Team
76fa71573b
feat(dnn/cuda): add cutlass nchw4 convolution
GitOrigin-RevId: 93c9b212f4
4 years ago
Megvii Engine Team
1f3f4abc38
fix(dnn): fix compile warnings
GitOrigin-RevId: 519a6c0c34
4 years ago
Megvii Engine Team
a558d4a253
fix(mgb/atlas): remove unnessary setdevice
GitOrigin-RevId: 7492de1f01
4 years ago
Megvii Engine Team
14ceda0dd3
ci(mge): fix Windows CI
symlink does not work on Windows
GitOrigin-RevId: 45505b8925
4 years ago
Megvii Engine Team
1fe8a21299
fix(mge): fix sublinear memory in jit.trace
GitOrigin-RevId: 190a330a8c
4 years ago
Megvii Engine Team
2df1ab9633
refactor(mge/jit): skip seed when checking equal rng op
GitOrigin-RevId: dae2086b36
4 years ago
Megvii Engine Team
5b6ebeb563
fix(mgb): append json file for dump and ready for midout open source
GitOrigin-RevId: 71ae7f1f4a
4 years ago
Megvii Engine Team
a81abc1d28
fix(mgb): append json file for dump and ready for midout open source
GitOrigin-RevId: 71ae7f1f4a
4 years ago