Megvii Engine Team
5b7ae26849
feat(mge): do not export F.nn.roi_pooling
GitOrigin-RevId: 8a07dd1da4
4 years ago
Megvii Engine Team
d22d1676e7
feat(mge): do not export F.nn.roi_align
GitOrigin-RevId: c2ac7bab32
4 years ago
Megvii Engine Team
9c5611345f
feat(mge): do not export F.nn.nms
GitOrigin-RevId: 7a0a328c37
4 years ago
Megvii Engine Team
56d624f5be
feat(mge): rename log_softmax -> logsoftmax
GitOrigin-RevId: 40ccfea5a7
4 years ago
Megvii Engine Team
164d9cdf1f
feat(mge): do not export F.nn.linear
GitOrigin-RevId: 589964729d
4 years ago
Megvii Engine Team
e42679b517
feat(mge): do not export F.nn.interpolate
GitOrigin-RevId: 031c6555c0
4 years ago
Megvii Engine Team
fd802e978f
feat(mge): do not export F.nn.embedding
GitOrigin-RevId: eb90506535
4 years ago
Megvii Engine Team
a640eeff62
feat(mge): remove batched_nms
GitOrigin-RevId: 01f9ee137c
4 years ago
Megvii Engine Team
205a39b07d
feat(mge): remove add_update
GitOrigin-RevId: 3068593eeb
4 years ago
Megvii Engine Team
67d33303ad
feat(mge): remove fast_tanh
GitOrigin-RevId: 7ad5f7ecce
4 years ago
Megvii Engine Team
78d2189441
feat(mge): rename comparison operators to their full name
GitOrigin-RevId: e503b038a1
4 years ago
Megvii Engine Team
e034874ae7
feat(mge): rename clamp -> clip
GitOrigin-RevId: 2b594a611e
4 years ago
Megvii Engine Team
e0da74852e
feat(opr): add confidential operator
GitOrigin-RevId: 53c2d4bc45
4 years ago
Megvii Engine Team
20cdd11d4d
docs(mge): fix docstring of cgtools
GitOrigin-RevId: ee8c93b9f7
4 years ago
Megvii Engine Team
2bd84d6789
feat(mge): add adaptive pooling python wrapper
GitOrigin-RevId: 789f1511ec
4 years ago
Megvii Engine Team
0e3037105c
feat(mge/module): add group_local_conv
GitOrigin-RevId: 0503377bc5
4 years ago
Megvii Engine Team
026af62042
docs(mge): docs typo fix
GitOrigin-RevId: 851f6de02f
4 years ago
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