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
912d733ea9
fix(dnn): support bool for IndexingMultiAxisVec
GitOrigin-RevId: ddcfaa06b0
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
7ac4dbc27d
fix(mgb/trace): finalize when exception raise
GitOrigin-RevId: b8ffd00a7e
4 years ago
Megvii Engine Team
2bd84d6789
feat(mge): add adaptive pooling python wrapper
GitOrigin-RevId: 789f1511ec
4 years ago
Megvii Engine Team
8109cc4bd1
fix(imperative): set exception which worker thread throws for rendezvous
GitOrigin-RevId: f583888fdf
4 years ago
Megvii Engine Team
2efba9a3a3
fix(mgb/test): use both rtol and atol for stable test result
GitOrigin-RevId: 82a1453e4a
4 years ago
Megvii Engine Team
0e3037105c
feat(mge/module): add group_local_conv
GitOrigin-RevId: 0503377bc5
4 years ago
Megvii Engine Team
36a4fb5611
ci(image): update docker image install python package
GitOrigin-RevId: 171e95b3d9
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
f63bca032c
test(mge/dtype): add quant and lowbit dtype test
GitOrigin-RevId: 97ca1a393a
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
477820fe5d
refactor(mge): apply python-module changes
GitOrigin-RevId: 2c769864c3
4 years ago
Megvii Engine Team
59a9275c66
fix(mge): fix optimize_for_inference during trace.dump
GitOrigin-RevId: e10f7c323a
4 years ago
Megvii Engine Team
bd7f885a2b
fix(mge/pytest): remove __init__ in py-test
GitOrigin-RevId: e16eae9cbc
4 years ago
Megvii Engine Team
06041f8a7e
fix(mge/functional): fix trace warp_perspective
GitOrigin-RevId: 2071bb63a8
4 years ago
Megvii Engine Team
92afcf3eb4
refactor(mge/imperative): move cgtools to megengine.utils and add load_and_inference in cgtools
GitOrigin-RevId: abfee3d4fa
4 years ago
Megvii Engine Team
66f2dbd753
feat(mge/imperative): add module __repr__ for imperative
GitOrigin-RevId: ac13cc4659
4 years ago
Megvii Engine Team
53ec6b833e
fix(mge): fix graph binding refcnt
GitOrigin-RevId: c955eb260a
4 years ago
Megvii Engine Team
1965276845
fix(mge): fix bool.sum()
GitOrigin-RevId: 62c482db40
4 years ago
Megvii Engine Team
42d3b92891
test(doctest): add doctest for megengine dir
GitOrigin-RevId: c31bbb1afc
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
fc74d5f2c3
feat(mge/dump): add more options for megbrain_graph.dump_graph
GitOrigin-RevId: 848e3c87c1
4 years ago
Megvii Engine Team
81b6a73382
fix(mge): fix bug of tensor.T
GitOrigin-RevId: 9fe9347b00
4 years ago
Megvii Engine Team
b111baf1b6
fix(mge/tensor): fix const target shape in reshape
GitOrigin-RevId: 7c04a9efba
4 years ago
Megvii Engine Team
715009b564
refactor(mge/api): remove external, dropout fix
GitOrigin-RevId: 5e6ff1a372
4 years ago
Megvii Engine Team
9005cf74d2
fix(mge/imperative): fix tracer leak
GitOrigin-RevId: 1a8ac20b46
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
91efd67dbf
refactor(mge/jit): change dump options, add test
GitOrigin-RevId: fbc0d51c2b
4 years ago
Megvii Engine Team
39d4328309
fix(mge/functional): functional api fixes
GitOrigin-RevId: fa206c4ff6
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
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