Megvii Engine Team
d225cbcdbe
feat(mge): add device name check
GitOrigin-RevId: d9910b6275
4 years ago
Megvii Engine Team
be2366423d
feat(sdk/load_and_run): add output-strip-info for dump with testcase imperative
GitOrigin-RevId: 337d95c7c2
4 years ago
Megvii Engine Team
026af62042
docs(mge): docs typo fix
GitOrigin-RevId: 851f6de02f
4 years ago
Megvii Engine Team
b2f9efb897
perf(mge): always async dispatch in jit.trace
GitOrigin-RevId: 6fa80b0dd9
4 years ago
Megvii Engine Team
36a4fb5611
ci(image): update docker image install python package
GitOrigin-RevId: 171e95b3d9
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
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
a398d4b5a5
chore(mge/functional): remove duplicated code
GitOrigin-RevId: f9efea46cb
4 years ago
Megvii Engine Team
477820fe5d
refactor(mge): apply python-module changes
GitOrigin-RevId: 2c769864c3
4 years ago
Megvii Engine Team
fe5649e456
fix(mge/imperative): remove duplicated opr
GitOrigin-RevId: 7d49785fad
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
f4860b9345
fix(mge/functional): fix trace topk
GitOrigin-RevId: c88ca8219b
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
1a24fb29c1
perf(mge/allreduce): put allreduce on another cuda stream
GitOrigin-RevId: 2e778dfa04
4 years ago
Megvii Engine Team
42d3b92891
test(doctest): add doctest for megengine dir
GitOrigin-RevId: c31bbb1afc
4 years ago
Megvii Engine Team
296a288594
fix(build/macos): fix macos build
GitOrigin-RevId: 66ec91c647
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
b7e9716098
feat(mge/plugin): add load_binary_tensor, copybara add comp_binary_iodump
GitOrigin-RevId: 8bb5d439d1
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
fc4c9292be
fix(mge/imperative): add missed __init__.py
GitOrigin-RevId: 0e863c8066
4 years ago
Megvii Engine Team
aac8de554e
fix(mge/parampacksplit): fix parampacksplit refcnt error
GitOrigin-RevId: c964465596
4 years ago
Megvii Engine Team
a7b9ece4a7
feat(mgb/comp_node): add set_prealloc_config
GitOrigin-RevId: e725e7efdd
4 years ago
Megvii Engine Team
066da0bff1
fix(mgb/comp_node): add lock_guard to sd.mtx & update warning message & add test
GitOrigin-RevId: ae0bf4b479
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
4ab259f558
refactor(mge): add some deprecation and migration guides
GitOrigin-RevId: 4793adf02b
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
6c1dbd40d9
docs(mge): fix doctest
GitOrigin-RevId: 131fed8733
4 years ago
Megvii Engine Team
e6715910c5
docs(mge/imperative): add docstring for GradManager
GitOrigin-RevId: 4c326206b8
4 years ago
Megvii Engine Team
91efd67dbf
refactor(mge/jit): change dump options, add test
GitOrigin-RevId: fbc0d51c2b
4 years ago
Megvii Engine Team
099ffeac12
chore(mge): typo and reword
GitOrigin-RevId: 3848da4824
4 years ago
Megvii Engine Team
39d4328309
fix(mge/functional): functional api fixes
GitOrigin-RevId: fa206c4ff6
4 years ago
Megvii Engine Team
e507228e74
feat(mge/examples): add distributed training examples using launcher
GitOrigin-RevId: 5db26f58eb
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
76f367962f
fix(mge/trace): fix op order in symbolic
GitOrigin-RevId: fbf081a199
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