Megvii Engine Team
4aaae9950b
feat(mge/functional): add python wrapper to resize opr
GitOrigin-RevId: b7cc6dd829
4 years ago
Megvii Engine Team
df976782fa
feat(mge/functional): add matinv
GitOrigin-RevId: d4fa8a8277
4 years ago
Megvii Engine Team
ea8eb4cf72
feat(mge/distributed): scalar support for distributed functions
GitOrigin-RevId: 53f3575baf
4 years ago
Megvii Engine Team
5c7d48cdb9
fix(mge/functional): fix tensor split
GitOrigin-RevId: 0a112ab0bd
4 years ago
Megvii Engine Team
4d75f691a0
feat(mge): restore remote send/recv
GitOrigin-RevId: 8b78fd5591
4 years ago
Megvii Engine Team
34c705fcbf
refactor(mge/imperative): move detach into C++
GitOrigin-RevId: 8c0d86cbbf
4 years ago
Megvii Engine Team
147cef52a0
refactor(mge/imperative): implement new tensor system
GitOrigin-RevId: 2dd4e460ac
4 years ago
Megvii Engine Team
9d928e7f83
refactor(mge/distributed): sync interpreter for distribtued launcher
GitOrigin-RevId: 8a88c272a1
4 years ago
Megvii Engine Team
ee4ea7fdc8
test(distributed/test): make distributed test more stronger
GitOrigin-RevId: 085fd1dcfd
4 years ago
Megvii Engine Team
442058aead
feat(mge/quantize): add batch matmul activation module for inference
GitOrigin-RevId: f6dbd1f4c0
4 years ago
Megvii Engine Team
82b0f67770
fix(mge/core): fix dtype promotion issue for quantized dtype
GitOrigin-RevId: 9d09c8fa6f
4 years ago
Megvii Engine Team
9cb3c07cd4
feat(mge/functional): add elemwise mode support string input
GitOrigin-RevId: 57be5cec7b
4 years ago
Megvii Engine Team
8fad00a158
feat(mge/functional/nn): add conv1d padding
GitOrigin-RevId: 1bbfd36b96
4 years ago
Megvii Engine Team
28dbadf729
feat(imperative): add remap opr
GitOrigin-RevId: 1fa7e18d4b
4 years ago
Megvii Engine Team
1217801133
perf(mge): add opdef for broadcast
GitOrigin-RevId: 92f0af29eb
4 years ago
Megvii Engine Team
dfc3935d06
fix(pytest/windows/impertive): fix windows wrong test skip
GitOrigin-RevId: ca0f04cfe7
4 years ago
Megvii Engine Team
215f88f373
fix(dnn/argmxx): fix argmxx on inf
GitOrigin-RevId: 740f67b73a
4 years ago
Megvii Engine Team
1fa143ce87
refactor(mge/functional): matmul supports symbolic shape, batched mv multiply
GitOrigin-RevId: c4d8cf3306
4 years ago
Megvii Engine Team
9748aebeea
refactor(mge): tensor_shape -> symbolic_shape
GitOrigin-RevId: 366dc048bf
4 years ago
Megvii Engine Team
0075e6acb0
refactor(mge): refactor cross_entropy
GitOrigin-RevId: 1fac5b5b14
4 years ago
Megvii Engine Team
d502e79f85
feat(mge): make F.norm numpy compatible
GitOrigin-RevId: 4e5236171c
4 years ago
Megvii Engine Team
aa62672629
feat(mge): make F.eye numpy compatible
GitOrigin-RevId: fee32537b4
4 years ago
Megvii Engine Team
d4e2d27213
feat(mge): do not export F.nn.conv_bias_activation
GitOrigin-RevId: e536f87fad
4 years ago
Megvii Engine Team
7d8a7e3efe
feat(mge): do not export F.loss.*
GitOrigin-RevId: a3ce4d3d76
4 years ago
Megvii Engine Team
5b2f0129cb
feat(mge): rename cross_entropy_with_softmax -> cross_entropy
GitOrigin-RevId: 9435c3260a
4 years ago
Megvii Engine Team
cf0507e1ba
feat(mge): remove F.identity
GitOrigin-RevId: 858be627ac
4 years ago
Megvii Engine Team
09241a1ff7
feat(mge): remove param_pack_* from functional
GitOrigin-RevId: a5fe25be8c
4 years ago
Megvii Engine Team
0d4568d6f8
feat(mge): rename broadcast -> broadcast_to
GitOrigin-RevId: 82f46ad2c2
4 years ago
Megvii Engine Team
1e2117f66d
feat(mge): remove add_axis and remove_axis
GitOrigin-RevId: 59611d43f9
4 years ago
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
e42679b517
feat(mge): do not export F.nn.interpolate
GitOrigin-RevId: 031c6555c0
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
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
57c4eccf3b
fix(mge/functional): add shape check for bc
GitOrigin-RevId: e152c1928c
4 years ago
Megvii Engine Team
2bd84d6789
feat(mge): add adaptive pooling python wrapper
GitOrigin-RevId: 789f1511ec
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
b80fade3f9
fix(mge/core): avoid create RawTensor with zero-stride numpy ndarray
GitOrigin-RevId: a9b2940bdc
4 years ago
Megvii Engine Team
bd7f885a2b
fix(mge/pytest): remove __init__ in py-test
GitOrigin-RevId: e16eae9cbc
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
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
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
44d0b5daf5
feat(imperative): enable to() to copy to device
GitOrigin-RevId: f9caf17d24
4 years ago