Megvii Engine Team
2070aaae09
fix(mge/functional): fix duplicate `transpose` in `__all__` and add `dimshuffle
GitOrigin-RevId: 2967088233
4 years ago
Megvii Engine Team
6ce796aec4
test(mge): replace `assertTensorClose` with `np.testing.assert_allclose`
GitOrigin-RevId: 2bf8c42cfd
4 years ago
Megvii Engine Team
a398d4b5a5
chore(mge/functional): remove duplicated code
GitOrigin-RevId: f9efea46cb
4 years ago
Megvii Engine Team
06041f8a7e
fix(mge/functional): fix trace warp_perspective
GitOrigin-RevId: 2071bb63a8
4 years ago
Megvii Engine Team
f4860b9345
fix(mge/functional): fix trace topk
GitOrigin-RevId: c88ca8219b
4 years ago
Megvii Engine Team
1a24fb29c1
perf(mge/allreduce): put allreduce on another cuda stream
GitOrigin-RevId: 2e778dfa04
4 years ago
Megvii Engine Team
b970c6daf1
fix(functional): hide format interface and skip test_conv_bias
GitOrigin-RevId: 7de75bf084
4 years ago
Megvii Engine Team
aac8de554e
fix(mge/parampacksplit): fix parampacksplit refcnt error
GitOrigin-RevId: c964465596
4 years ago
Megvii Engine Team
715009b564
refactor(mge/api): remove external, dropout fix
GitOrigin-RevId: 5e6ff1a372
4 years ago
Megvii Engine Team
70bc746c20
fix(mge): fix bug of identity()
GitOrigin-RevId: 4bfd3cafb5
4 years ago
Megvii Engine Team
a8c75ee530
refactor(mge/functional): make docs better
GitOrigin-RevId: acb287d48b
4 years ago
Megvii Engine Team
6c1dbd40d9
docs(mge): fix doctest
GitOrigin-RevId: 131fed8733
4 years ago
Megvii Engine Team
39d4328309
fix(mge/functional): functional api fixes
GitOrigin-RevId: fa206c4ff6
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
5474b000a7
fix(mge/functional): fix convert_inputs before apply
GitOrigin-RevId: ab41974a1f
4 years ago
Megvii Engine Team
e9104ef157
fix(mge/parampack): fix copy stream, import cycle
GitOrigin-RevId: 673e11c5b6
4 years ago
Megvii Engine Team
b5016b9d29
feat(mge/parampack): add parampack in allreduce callback
GitOrigin-RevId: 73d53eeba1
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
8dc23e0fdf
fix(mge/functional): fix indexing_one_hot and remote_recv
GitOrigin-RevId: 00bdfb502b
4 years ago
Megvii Engine Team
0b88ec3cff
refactor(mge/imperative): use apply.super in distributed
GitOrigin-RevId: e53276bd46
4 years ago
Megvii Engine Team
5474538f39
refactor(mge/imperative): fork multipledispatch
GitOrigin-RevId: a7c25a4302
4 years ago
Megvii Engine Team
03320a0548
fix(functional): change to user-friendly nms api and fix concat
GitOrigin-RevId: 76b4190270
4 years ago
Megvii Engine Team
a226f02e7a
fix(mge/imperative): fix syncbn in symbolic mode
GitOrigin-RevId: a9794318a7
4 years ago
Megvii Engine Team
236c7ec80b
refactor(imperative): remove unused opdef gen
GitOrigin-RevId: 509dbd0bb2
4 years ago
Megvii Engine Team
0067dcf068
test(mge/imperative): add module test to imperative
GitOrigin-RevId: e690bc42b0
4 years ago
Megvii Engine Team
7fadc16d3c
refactor(mge/functional): support tensor shape in interpolate and split
GitOrigin-RevId: 6430b64f01
4 years ago
Megvii Engine Team
60076f4715
test(mge/parampack): add parampack related opr test
GitOrigin-RevId: 550c6b6443
4 years ago
Megvii Engine Team
d1bbbcc6a4
docs(mge/functional): add doc for copy, param pack split and concat
GitOrigin-RevId: c93bdae290
4 years ago
Megvii Engine Team
7e8f720953
refactor(mge/tensor): tensor reduce supports keepdims
GitOrigin-RevId: 8ed95e0fb8
4 years ago
Megvii Engine Team
40d18c8920
fix(mge/imperative): fix tests when shape is tensor
GitOrigin-RevId: fd0095c1ec
4 years ago
Megvii Engine Team
f4927db2fb
feat(mge/functional): support where func
GitOrigin-RevId: 9df6421ebe
4 years ago
Megvii Engine Team
56cb5d6a16
fix(mge/functional): int operations(div, exp, pow)
GitOrigin-RevId: dc434fa7ec
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