Megvii Engine Team
9f4bffbd00
fix(mge/tensor): fix valid_broadcast
GitOrigin-RevId: 562b7664e2
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
205a39b07d
feat(mge): remove add_update
GitOrigin-RevId: 3068593eeb
4 years ago
Megvii Engine Team
20cdd11d4d
docs(mge): fix docstring of cgtools
GitOrigin-RevId: ee8c93b9f7
4 years ago
Megvii Engine Team
57c4eccf3b
fix(mge/functional): add shape check for bc
GitOrigin-RevId: e152c1928c
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
2f4a75e7f2
feat(mge/utils): redesign dtype promotion
GitOrigin-RevId: 4f2fe1b6ce
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
36a4fb5611
ci(image): update docker image install python package
GitOrigin-RevId: 171e95b3d9
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
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
1965276845
fix(mge): fix bool.sum()
GitOrigin-RevId: 62c482db40
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
6c1dbd40d9
docs(mge): fix doctest
GitOrigin-RevId: 131fed8733
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
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
478f2c51dc
feat(mge/jit): add trace/dump options
GitOrigin-RevId: 0f43c14599
4 years ago
Megvii Engine Team
67859f04e1
fix(imperative): add __array__ and __array_wrap__ for tensorwrapper
GitOrigin-RevId: 87d4ab6c8e
4 years ago
Megvii Engine Team
5aa19f3d4a
test(mge/imperative): add more testcases for function
GitOrigin-RevId: e0675994f1
4 years ago
Megvii Engine Team
fc3eca845b
feat(mge/imperative/jit): add dump input shape and xornet example
GitOrigin-RevId: 5e2acd4052
4 years ago
Megvii Engine Team
1b2194cdce
feat(whl/internal): put imperative rt to megbrain
GitOrigin-RevId: f3b6e492d7
4 years ago
Megvii Engine Team
76dbaa2741
feat(mge/imperative): add name, make_h2d, dump_graph to graph runtime
GitOrigin-RevId: b8681a31a8
4 years ago
Megvii Engine Team
e6e29748b9
chore(mge/imperative): fix Graph.make_const
GitOrigin-RevId: 0f4c62aebf
4 years ago
Megvii Engine Team
ac3408bfaa
chore(mge): add VarNode.value
GitOrigin-RevId: 1dc0d0c711
4 years ago
Megvii Engine Team
4101d5bca0
refactor(mge/imperative): add BackwardGraph.interpret
GitOrigin-RevId: bb3a59380e
4 years ago
Megvii Engine Team
d4bad71190
feat(mge): add jit.trace
GitOrigin-RevId: ec647324c0
4 years ago
Megvii Engine Team
332959003f
chore(mge): add multipledispatch license
GitOrigin-RevId: 502c53324c
4 years ago
Megvii Engine Team
5474538f39
refactor(mge/imperative): fork multipledispatch
GitOrigin-RevId: a7c25a4302
4 years ago
Megvii Engine Team
ad9ac521ff
refactor(mge/imperative): remove abandoned code
GitOrigin-RevId: 0178bb5684
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
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
92f7cceb11
fix(mge/functional): fix cond_take when index is numpy array
GitOrigin-RevId: 5fb93740f4
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