Megvii Engine Team
|
c294b9d18b
|
refactor(mge/tensor): remove old implementation
remove core.tensor, raw_tensor,TensorWrapper
avoid create tensor with zero-stride numpy ndarray
GitOrigin-RevId: 4fe5c4c5ba
|
4 years ago |
Megvii Engine Team
|
147cef52a0
|
refactor(mge/imperative): implement new tensor system
GitOrigin-RevId: 2dd4e460ac
|
4 years ago |
Megvii Engine Team
|
2ad8c5e1e9
|
fix(mge/io_remote): fix remote send/recv gradient at trace
GitOrigin-RevId: 7886efd0c1
|
4 years ago |
Megvii Engine Team
|
94dba16ff4
|
perf(mge/imperative): misc optimizations
GitOrigin-RevId: bbe7a10b00
|
4 years ago |
Megvii Engine Team
|
094601e834
|
feat(mge/distributed): allow remote grad by using grad manager
GitOrigin-RevId: a890c206a5
|
4 years ago |
Megvii Engine Team
|
fccb2510c4
|
perf(mge): add opdef for elemwise and batchnorm
GitOrigin-RevId: d51fad9867
|
4 years ago |
Megvii Engine Team
|
9005cf74d2
|
fix(mge/imperative): fix tracer leak
GitOrigin-RevId: 1a8ac20b46
|
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
|
9faa32fcb7
|
fix(mge/imperative): fix grad callback
GitOrigin-RevId: 6f843b0106
|
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
|
5474538f39
|
refactor(mge/imperative): fork multipledispatch
GitOrigin-RevId: a7c25a4302
|
4 years ago |
Megvii Engine Team
|
6b380e8965
|
feat(mge/imperative): run oss test and restore cmake list build items
GitOrigin-RevId: 11411b6964
|
4 years ago |